diff options
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 |