Owos commited on
Commit
cace7f6
1 Parent(s): ed0c6e6

Update neural_style.py

Browse files
Files changed (1) hide show
  1. 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='lbfgs')
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