Don't checkpoint so often

JasonGross · JasonGross · commit fa05c1c916ab · 2025-01-26T19:43:54.000-08:00
diff --git a/gbmi/exp_modular_arithmetic/run_train_clocks.py b/gbmi/exp_modular_arithmetic/run_train_clocks.py
@@ -15,6 +15,7 @@
                 {
                     "seed": seed,
                     "train_for": (10000, "epochs"),
+                    "checkpoint_every": (2000, "epochs"),
                     ("experiment", "logging_options"): ModelMatrixLoggingOptions.none(),
                 },
                 post_init=True,

Original file line number	Diff line number	Diff line change
`@@ -15,6 +15,7 @@`
`15`	`15`	`{`
`16`	`16`	`"seed": seed,`
`17`	`17`	`"train_for": (10000, "epochs"),`
	`18`	`+ "checkpoint_every": (2000, "epochs"),`
`18`	`19`	`("experiment", "logging_options"): ModelMatrixLoggingOptions.none(),`
`19`	`20`	`},`
`20`	`21`	`post_init=True,`