From 6e5f2d1a62f4f3bf0e87829082b2120ca440ddf0 Mon Sep 17 00:00:00 2001
From: Logan Bauman <logan_bauman@brown.edu>
Date: Thu, 5 May 2022 11:12:46 -0400
Subject: hi

---
 losses.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

(limited to 'losses.py')

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()
-- 
cgit v1.2.3-70-g09d2