Update train.py

jakedubb · jakedubb · commit 654da1c23a2b · 2019-11-12T16:24:06.000-08:00
diff --git a/code/training/train.py b/code/training/train.py
@@ -69,7 +69,7 @@
 print("Running train.py")
 
 # Randomly pic alpha
-alphas = np.arange(0.1, 0.2, 0.3)
+alphas = np.arange(0.0, 0.5, 0.01)
 alpha = alphas[np.random.choice(alphas.shape[0], 1, replace=False)][0]
 print(alpha)
 run.log("alpha", alpha)
