Spaces:
Runtime error
Runtime error
Update neural_style.py
Browse files- neural_style.py +1 -1
neural_style.py
CHANGED
@@ -25,7 +25,7 @@ parser.add_argument("-tv_weight", type=float, default=1e-3)
|
|
25 |
parser.add_argument("-num_iterations", type=int, default=1000)
|
26 |
parser.add_argument("-init", choices=['random', 'image'], default='random')
|
27 |
parser.add_argument("-init_image", default=None)
|
28 |
-
parser.add_argument("-optimizer", choices=['lbfgs', 'adam'], default='
|
29 |
parser.add_argument("-learning_rate", type=float, default=1e0)
|
30 |
parser.add_argument("-lbfgs_num_correction", type=int, default=100)
|
31 |
|
|
|
25 |
parser.add_argument("-num_iterations", type=int, default=1000)
|
26 |
parser.add_argument("-init", choices=['random', 'image'], default='random')
|
27 |
parser.add_argument("-init_image", default=None)
|
28 |
+
parser.add_argument("-optimizer", choices=['lbfgs', 'adam'], default='adam')
|
29 |
parser.add_argument("-learning_rate", type=float, default=1e0)
|
30 |
parser.add_argument("-lbfgs_num_correction", type=int, default=100)
|
31 |
|