Spaces:
Runtime error
Runtime error
Update neural_style.py
Browse files- neural_style.py +3 -3
neural_style.py
CHANGED
|
@@ -77,13 +77,13 @@ class TransferParams():
|
|
| 77 |
style_scale = 1.0
|
| 78 |
original_colors = 0
|
| 79 |
pooling = 'max'
|
| 80 |
-
model_file = 'models/nin_imagenet.pth'
|
| 81 |
disable_check = False
|
| 82 |
backend = 'mkl'
|
| 83 |
cudnn_autotune = False
|
| 84 |
seed = -1
|
| 85 |
-
content_layers = 'relu0,relu3,relu7,relu12'
|
| 86 |
-
style_layers = 'relu0,relu3,relu7,relu12'
|
| 87 |
multidevice_strategy = '4,7,29'
|
| 88 |
|
| 89 |
def main():
|
|
|
|
| 77 |
style_scale = 1.0
|
| 78 |
original_colors = 0
|
| 79 |
pooling = 'max'
|
| 80 |
+
model_file = 'models/vgg16-00b39a1b.pth'#nin_imagenet.pth'
|
| 81 |
disable_check = False
|
| 82 |
backend = 'mkl'
|
| 83 |
cudnn_autotune = False
|
| 84 |
seed = -1
|
| 85 |
+
content_layers = 'relu4_2'#relu0,relu3,relu7,relu12'
|
| 86 |
+
style_layers = 'relu1_1,relu2_1,relu3_1,relu4_1,relu5_1'#relu0,relu3,relu7,relu12'
|
| 87 |
multidevice_strategy = '4,7,29'
|
| 88 |
|
| 89 |
def main():
|