diff options
| -rw-r--r-- | hyperparameters.py | 4 | ||||
| -rw-r--r-- | losses.py | 4 |
2 files changed, 6 insertions, 2 deletions
diff --git a/hyperparameters.py b/hyperparameters.py index f59b9747..8a3da6e2 100644 --- a/hyperparameters.py +++ b/hyperparameters.py @@ -19,3 +19,7 @@ optimizer is used. Refer to the default learning rate parameter learning_rate = 1e-4 momentum = 0.01 + +alpha = 1e-2 + +beta = 1e-5 @@ -17,8 +17,8 @@ class YourModel(tf.keras.Model): #perhaps consider cropping to avoid distortion self.style_image = transform.resize(style_image, np.shape(style_image), anti_aliasing=True) self.x = tf.Variable(tf.random.uniform(np.shape(content_image)), trainable=True) - self.alpha = 1 - self.beta = 1 + self.alpha = hp.alpha + self.beta = hp.beta self.optimizer = tf.keras.optimizers.RMSprop(learning_rate=hp.learning_rate, momentum=hp.momentum) |