Fix clock training

JasonGross · JasonGross · commit b8a1c335ed1c · 2025-01-26T19:29:12.000-08:00
diff --git a/gbmi/exp_modular_arithmetic/run_train_clocks.py b/gbmi/exp_modular_arithmetic/run_train_clocks.py
@@ -14,7 +14,7 @@
                 CLOCK_CONFIG,
                 {
                     "seed": seed,
-                    ("experiment", "train_for"): (10000, "epochs"),
+                    "train_for": (10000, "epochs"),
                     ("experiment", "logging_options"): ModelMatrixLoggingOptions.none(),
                 },
                 post_init=True,

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