diff options
| author | Logan Bauman <logan_bauman@brown.edu> | 2022-05-04 15:42:18 -0400 |
|---|---|---|
| committer | Logan Bauman <logan_bauman@brown.edu> | 2022-05-04 15:42:18 -0400 |
| commit | 6d4c7a0608a91e3a94d1b174aa9094616c778cb6 (patch) | |
| tree | 207e60501bbda7c6a2297b6427c141e4571ace91 /losses.py | |
| parent | df0a9240bac34d2bda0d3c7c836dbce2ce781344 (diff) | |
add alpha and beta hyper parameters
Diffstat (limited to 'losses.py')
| -rw-r--r-- | losses.py | 4 |
1 files changed, 2 insertions, 2 deletions
@@ -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) |