diff options
| author | Rowan Smith <r.smith@internode.on.net> | 2017-02-26 00:11:21 +1100 |
|---|---|---|
| committer | Rowan Smith <r.smith@internode.on.net> | 2017-02-26 00:11:21 +1100 |
| commit | 86f95070c0d3d00ebe133caa823fdb26e28baa59 (patch) | |
| tree | 41c875a3c7447cb4134f485dd21d4fe516db9d7f /neural_style.py | |
| parent | bc216125c60b902b767bc3ccd97fb49bb3b55580 (diff) | |
update deprecated tf.initialize_all_variables to tf.global_variables_initializer
Diffstat (limited to 'neural_style.py')
| -rw-r--r-- | neural_style.py | 4 |
1 files 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 |