Update train.py

sayakpaul · Jan 9, 2020 · 353e336 · 353e336
1 parent 952d94c
commit 353e336
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/Bayesian_Search_W&B/train.py b/Bayesian_Search_W&B/train.py
@@ -37,8 +37,7 @@
 configs = {
     'layers': 128,
     'batch_size': 64,
-    'epochs': 5,
-    'method': METHOD
+    'epochs': 5
 }
 
 # Initilize a new wandb run
@@ -74,4 +73,4 @@
          batch_size=config.batch_size,
          validation_data=(X_test, y_test),
          callbacks=[WandbCallback(data_type="image", 
-            validation_data=(X_test, y_test), labels=labels)])
+            validation_data=(X_test, y_test), labels=labels)])