1 files changed, 1 insertions, 1 deletions
diff --git a/losses.py b/losses.py
index 407412a1..7c99d73a 100644
--- a/losses.py
+++ b/losses.py
@@ -21,7 +21,7 @@ class YourModel(tf.keras.Model):
 
         print(self.content_image.shape, self.style_image.shape)
 
-        self.optimizer = tf.keras.optimizers.RMSprop(learning_rate=hp.learning_rate, momentum=hp.momentum)
+        self.optimizer = tf.keras.optimizers.Adam()
 
         self.vgg16 = [
             # Block 1