hi

1 files changed, 3 insertions, 1 deletions

diff --git a/losses.py b/losses.py
index 7c99d73a..7198ebf4 100644
--- a/losses.py
+++ b/losses.py
@@ -11,7 +11,7 @@ class YourModel(tf.keras.Model):
        
         self.content_image = transform.resize(content_image, tf.shape(style_image), anti_aliasing=True)
         self.content_image = tf.expand_dims(self.content_image, axis=0)
-
+        
         #perhaps consider cropping to avoid distortion
         self.style_image = transform.resize(style_image, tf.shape(style_image), anti_aliasing=True)
         self.style_image = tf.expand_dims(self.style_image, axis=0)
@@ -19,6 +19,8 @@ class YourModel(tf.keras.Model):
         self.alpha = hp.alpha
         self.beta = hp.beta
 
+        print(self.x.shape)
+
         print(self.content_image.shape, self.style_image.shape)
 
         self.optimizer = tf.keras.optimizers.Adam()