Owos commited on
Commit
ed0c6e6
1 Parent(s): 25453dd

changed optimizer to adam

Browse files
Files changed (1) hide show
  1. neural_style.py +1 -1
neural_style.py CHANGED
@@ -67,7 +67,7 @@ class TransferParams():
67
  num_iterations = 1000
68
  init = 'random'
69
  init_image = None
70
- optimizer = 'lbfgs'
71
  learning_rate = 1e0
72
  lbfgs_num_correction = 100
73
  print_iter = 50
 
67
  num_iterations = 1000
68
  init = 'random'
69
  init_image = None
70
+ optimizer = 'adam'
71
  learning_rate = 1e0
72
  lbfgs_num_correction = 100
73
  print_iter = 50