diff options
Diffstat (limited to 'losses.py')
-rw-r--r-- | losses.py | 2 |
1 files changed, 1 insertions, 1 deletions
@@ -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 |