wip

wandb · Feb 26, 2024 · 8027d64 · 8027d64
1 parent a9c4bcd
commit 8027d64
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/jobs/fashion_mnist_train/job.py b/jobs/fashion_mnist_train/job.py
@@ -28,7 +28,7 @@ def train(project: Optional[str], entity: Optional[str], **kwargs: Any):
     # get training parameters from config
     epochs = train_config.get("epochs", 10)
     learning_rate = train_config.get("learning_rate", 0.001)
-
+    steps_per_epoch = train_config.get("steps_per_epoch", 10)
 
     # load data
     (train_X, train_y), (test_X, test_y) = fashion_mnist.load_data()