diff options
| author | Michael Foiani <sotech117@michaels-mbp-4.devices.brown.edu> | 2022-05-09 02:02:42 -0400 |
|---|---|---|
| committer | Michael Foiani <sotech117@michaels-mbp-4.devices.brown.edu> | 2022-05-09 02:02:42 -0400 |
| commit | c8386be631f38cde31391124423f77b7479d1bad (patch) | |
| tree | 3615b99b52fa2f620bc88118bf4aee9e697361bf /hyperparameters.py | |
| parent | 13826824ec58e85557efb3a12fb0456ffb20e46d (diff) | |
added examples and fixed the io for checkpoints
Diffstat (limited to 'hyperparameters.py')
| -rw-r--r-- | hyperparameters.py | 16 |
1 files changed, 14 insertions, 2 deletions
diff --git a/hyperparameters.py b/hyperparameters.py index 0ca70764..1d4c0e8d 100644 --- a/hyperparameters.py +++ b/hyperparameters.py @@ -18,12 +18,24 @@ optimizer is used. Refer to the default learning rate parameter """ learning_rate = .002 +""" +Beta_1 is the first hyperparameter for the Adam optimizer. +""" beta_1 = .99 +""" +epsilon for the Adam optimizer. +""" epsilon = 1e-1 -momentum = 0.01 - +""" +A critical parameter for style transfer. The value for this will determine +how much the generated image is "influenced" by the CONTENT image. +""" alpha = .05 +""" +A critical parameter for style transfer. The value for this will determine +how much the generated image is "influenced" by the STYLE image. +""" beta = 5 |