diff options
Diffstat (limited to 'losses.py')
-rw-r--r-- | losses.py | 2 |
1 files changed, 1 insertions, 1 deletions
@@ -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 |