diff options
Diffstat (limited to 'hyperparameters.py')
-rw-r--r-- | hyperparameters.py | 8 |
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 |