Spaces:

AfrodreamsAI
/

afrodreams

Runtime error

Owos commited on Sep 26, 2022

Commit

4e5c133

•

1 Parent(s): ca8ddf3

Update neural_style.py

Files changed (1) hide show

neural_style.py CHANGED Viewed

@@ -82,8 +82,8 @@ class TransferParams():
     backend = 'mkl'
     cudnn_autotune = False
     seed = -1
-    content_layers = 'relu4_2'
-    style_layers = 'relu1_1,relu2_1,relu3_1,relu4_1,relu5_1'
     multidevice_strategy = '4,7,29'
 def main():

     backend = 'mkl'
     cudnn_autotune = False
     seed = -1
+    content_layers = 'relu0,relu3,relu7,relu12'
+    style_layers = 'relu0,relu3,relu7,relu12'
     multidevice_strategy = '4,7,29'
 def main():