diff options
Diffstat (limited to 'hyperparameters.py')
| -rw-r--r-- | hyperparameters.py | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/hyperparameters.py b/hyperparameters.py index a15d04ac..75528742 100644 --- a/hyperparameters.py +++ b/hyperparameters.py @@ -9,17 +9,17 @@ 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 = 10000 +num_epochs = 500 """ A critical parameter that can dramatically affect whether training succeeds or fails. The value for this depends significantly on which optimizer is used. Refer to the default learning rate parameter """ -learning_rate = .002 +learning_rate = 1e2 momentum = 0.01 -alpha = 1e-2 +alpha = 1 -beta = 5e1 +beta = 100 |