aboutsummaryrefslogtreecommitdiff
path: root/losses.py
diff options
context:
space:
mode:
Diffstat (limited to 'losses.py')
-rw-r--r--losses.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/losses.py b/losses.py
index 407412a1..8c142f86 100644
--- a/losses.py
+++ b/losses.py
@@ -145,7 +145,7 @@ class YourModel(tf.keras.Model):
for i in range(len(art_layers)):
art_layer = art_layers[i]
input_layer = input_layers[i]
- L_style = tf.math.add(L_style, self.layer_loss(art_layer, input_layer))
+ L_style = tf.math.add(L_style, (1/5)*self.layer_loss(art_layer, input_layer))
#print('style loss', L_style)
return L_style