aboutsummaryrefslogtreecommitdiff
path: root/hyperparameters.py
diff options
context:
space:
mode:
authorDavid Doan <daviddoan@davids-mbp-3.devices.brown.edu>2022-05-07 18:55:47 -0400
committerDavid Doan <daviddoan@davids-mbp-3.devices.brown.edu>2022-05-07 18:55:47 -0400
commit79e999da9c3eff40006267b53bab0193242322df (patch)
tree3456585b93aa57b5010c3ecee5ff785f547be753 /hyperparameters.py
parent7a03723ffb9ac7613f3c5537a4d5c7cb9a4a3f3a (diff)
changes
Diffstat (limited to 'hyperparameters.py')
-rw-r--r--hyperparameters.py8
1 files changed, 5 insertions, 3 deletions
diff --git a/hyperparameters.py b/hyperparameters.py
index 75528742..ac2beda8 100644
--- a/hyperparameters.py
+++ b/hyperparameters.py
@@ -9,7 +9,7 @@ Number of epochs. If you experiment with more complex networks you
might need to increase this. Likewise if you add regularization that
slows training.
"""
-num_epochs = 500
+num_epochs = 200
"""
A critical parameter that can dramatically affect whether training
@@ -20,6 +20,8 @@ learning_rate = 1e2
momentum = 0.01
-alpha = 1
+alpha = .05
+beta = 5
+# alpha = 1e-5
-beta = 100
+# beta = 1e-2