Owos commited on
Commit
5dab46e
·
1 Parent(s): fe48878

Update neural_style.py

Browse files
Files changed (1) hide show
  1. neural_style.py +3 -3
neural_style.py CHANGED
@@ -64,14 +64,14 @@ class TransferParams():
64
  style_weight = 1e2
65
  normalize_weights = False
66
  tv_weight = 1e-3
67
- num_iterations = 500
68
  init = 'random'
69
  init_image = None
70
  optimizer = 'lbfgs'
71
  learning_rate = 1e0
72
  lbfgs_num_correction = 100
73
  print_iter = 50
74
- save_iter = 500
75
  output_image = 'out.png'
76
  log_level = 10
77
  style_scale = 1.0
@@ -251,7 +251,7 @@ def transfer(params):
251
 
252
  final_image = ''
253
  def maybe_save(t):
254
- should_save = params.save_iter > 450 and t % params.save_iter == 0
255
  should_save = should_save or t == params.num_iterations
256
  if should_save:
257
  output_filename, file_extension = os.path.splitext(params.output_image)
 
64
  style_weight = 1e2
65
  normalize_weights = False
66
  tv_weight = 1e-3
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
74
+ save_iter = 1000
75
  output_image = 'out.png'
76
  log_level = 10
77
  style_scale = 1.0
 
251
 
252
  final_image = ''
253
  def maybe_save(t):
254
+ should_save = params.save_iter > 950 and t % params.save_iter == 0
255
  should_save = should_save or t == params.num_iterations
256
  if should_save:
257
  output_filename, file_extension = os.path.splitext(params.output_image)