From 4c48370de02f5db3681bd35228c69faafe3df57a Mon Sep 17 00:00:00 2001 From: Logan Bauman Date: Wed, 4 May 2022 15:07:10 -0400 Subject: fix loss_fn --- losses.py | 6 +++--- 1 file 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) -- cgit v1.2.3-70-g09d2