Spaces:

AfrodreamsAI
/

afrodreams

Runtime error

Owos commited on Nov 25, 2022

Commit

1dfc71d

•

1 Parent(s): c71ec65

Update neural_style.py

Files changed (1) hide show

neural_style.py CHANGED Viewed

@@ -77,13 +77,13 @@ class TransferParams():
     style_scale = 1.0
     original_colors = 0
     pooling = 'max'
-    model_file = 'models/vgg16-00b39a1b.pth'#nin_imagenet.pth'
     disable_check = False
     backend = 'mkl'
     cudnn_autotune = False
     seed = -1
-    content_layers = 'relu4_2'#relu0,relu3,relu7,relu12'
-    style_layers = 'relu1_1,relu2_1,relu3_1,relu4_1,relu5_1'#relu0,relu3,relu7,relu12'
     multidevice_strategy = '4,7,29'
 def main():

     style_scale = 1.0
     original_colors = 0
     pooling = 'max'
+    model_file = 'models/nin_imagenet.pth'#vgg16-00b39a1b.pth'
     disable_check = False
     backend = 'mkl'
     cudnn_autotune = False
     seed = -1
+    content_layers = 'relu0,relu3,relu7,relu12'#relu4_2'#
+    style_layers = 'relu0,relu3,relu7,relu12'#relu1_1,relu2_1,relu3_1,relu4_1,relu5_1'#'
     multidevice_strategy = '4,7,29'
 def main():