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