wip

wandb · Jan 16, 2024 · d824adc · d824adc
1 parent d57286c
commit d824adc
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/.circleci/config.yml b/.circleci/config.yml
@@ -47,3 +47,7 @@ workflows:
                   "jobs/sweep_schedulers/optuna_scheduler",
                   "jobs/sweep_schedulers/wandb_scheduler",
                 ]
+        filters:
+          branches:
+            only:
+              - main
diff --git a/jobs/fashion_mnist_train/job.py b/jobs/fashion_mnist_train/job.py
@@ -117,10 +117,8 @@ def model_arch():
     # size with a kernel size of 2x2
     models.add(MaxPooling2D(pool_size=(2, 2)))
     models.add(Conv2D(128, (5, 5), padding="same", activation="relu"))
-
     models.add(MaxPooling2D(pool_size=(2, 2)))
     models.add(Conv2D(256, (5, 5), padding="same", activation="relu"))
-
     models.add(MaxPooling2D(pool_size=(2, 2)))
 
     # Once the convolutional and pooling