fix loss_fn

1 files changed, 3 insertions, 3 deletions

diff --git a/losses.py b/losses.py
index caad3067..0128749a 100644
--- a/losses.py
+++ b/losses.py
@@ -84,9 +84,9 @@ class YourModel(tf.keras.Model):
 
 
     def loss_fn(self, p, a, x):
-        _, photo_layers = self.forward_pass(p)
-        _, art_layers = self.forward_pass(a)
-        _, input_layers = self.forward_pass(x)
+        _, photo_layers = self.call(p)
+        _, art_layers = self.call(a)
+        _, input_layers = self.call(x)
 
         content_l = self.content_loss(photo_layers, input_layers)
         style_l = self.style_loss(art_layers, input_layers)