From 9af705f1786cdc157cf2bfb09aa8f95dab3f1122 Mon Sep 17 00:00:00 2001 From: Cameron Date: Sat, 8 Oct 2016 16:19:49 -0700 Subject: Fixed 'verbose' error --- neural_style.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/neural_style.py b/neural_style.py index 797ffa5..74badee 100644 --- a/neural_style.py +++ b/neural_style.py @@ -591,13 +591,12 @@ def minimize_with_adam(sess, net, optimizer, init_img): iterations += 1 def get_optimizer(loss): - v = 1 if args.verbose else 0 if args.optimizer == 'lbfgs': optimizer = tf.contrib.opt.ScipyOptimizerInterface( loss, method='L-BFGS-B', - options={'maxiter': args.max_iterations - 'disp': v}) + options={'maxiter': args.max_iterations, + 'disp': args.verbose}) elif args.optimizer == 'adam': optimizer = tf.train.AdamOptimizer(args.learning_rate) return optimizer -- cgit v1.2.3-70-g09d2