diff options
| author | Logan Bauman <logan_bauman@brown.edu> | 2022-05-07 08:15:54 -0400 |
|---|---|---|
| committer | Logan Bauman <logan_bauman@brown.edu> | 2022-05-07 08:15:54 -0400 |
| commit | 46a7929942f90d960a0a4d6e35251c900bd45fb2 (patch) | |
| tree | d5933713cc33b75e3b7315ecdbd2a91d6b02804e /losses.py | |
| parent | 98be0e58a000880d7e05e79f977452642eab54c6 (diff) | |
patch preprocessing, fix layer loss and hyper params
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 |