aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorLogan Bauman <logan_bauman@brown.edu>2022-05-04 15:42:18 -0400
committerLogan Bauman <logan_bauman@brown.edu>2022-05-04 15:42:18 -0400
commit6d4c7a0608a91e3a94d1b174aa9094616c778cb6 (patch)
tree207e60501bbda7c6a2297b6427c141e4571ace91
parentdf0a9240bac34d2bda0d3c7c836dbce2ce781344 (diff)
add alpha and beta hyper parameters
-rw-r--r--hyperparameters.py4
-rw-r--r--losses.py4
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
diff --git a/losses.py b/losses.py
index fd68e199..542aa144 100644
--- a/losses.py
+++ b/losses.py
@@ -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)