From 86f95070c0d3d00ebe133caa823fdb26e28baa59 Mon Sep 17 00:00:00 2001 From: Rowan Smith Date: Sun, 26 Feb 2017 00:11:21 +1100 Subject: update deprecated tf.initialize_all_variables to tf.global_variables_initializer --- neural_style.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/neural_style.py b/neural_style.py index be5a95e..0fa4f13 100644 --- a/neural_style.py +++ b/neural_style.py @@ -617,7 +617,7 @@ def stylize(content_img, style_imgs, init_img, frame=None): def minimize_with_lbfgs(sess, net, optimizer, init_img): if args.verbose: print('\nMINIMIZING LOSS USING: L-BFGS OPTIMIZER') - init_op = tf.initialize_all_variables() + init_op = tf.global_variables_initializer() sess.run(init_op) sess.run(net['input'].assign(init_img)) optimizer.minimize(sess) @@ -625,7 +625,7 @@ def minimize_with_lbfgs(sess, net, optimizer, init_img): def minimize_with_adam(sess, net, optimizer, init_img, loss): if args.verbose: print('\nMINIMIZING LOSS USING: ADAM OPTIMIZER') train_op = optimizer.minimize(loss) - init_op = tf.initialize_all_variables() + init_op = tf.global_variables_initializer() sess.run(init_op) sess.run(net['input'].assign(init_img)) iterations = 0 -- cgit v1.2.3-70-g09d2