aboutsummaryrefslogtreecommitdiff
path: root/losses.py
diff options
context:
space:
mode:
authorLogan Bauman <logan_bauman@brown.edu>2022-05-07 14:16:42 -0400
committerLogan Bauman <logan_bauman@brown.edu>2022-05-07 14:16:42 -0400
commit1b54bfa9de44f29ca2046d1eeb0ab174ae0dadbd (patch)
tree5b9738733d10bd36e00e6d5f7a6463174f52059b /losses.py
parent95be7aba1fa85881253a0a09752204863f31bcb8 (diff)
hi
Diffstat (limited to 'losses.py')
-rw-r--r--losses.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/losses.py b/losses.py
index eba94c36..f08a734c 100644
--- a/losses.py
+++ b/losses.py
@@ -34,7 +34,7 @@ class YourModel(tf.keras.Model):
#print(self.content_image.shape, self.style_image.shape)
- self.optimizer = tf.keras.optimizers.Adam()
+ self.optimizer = tf.keras.optimizers.Adam(hp.learning_rate)
self.vgg16 = [
# Block 1