diff options
| author | Cameron <cysmith1010@gmail.com> | 2017-02-27 14:08:43 -0700 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2017-02-27 14:08:43 -0700 |
| commit | e515f89f3635f59a9f96bef2a393b3a11b0a90f1 (patch) | |
| tree | 41c875a3c7447cb4134f485dd21d4fe516db9d7f | |
| parent | bc216125c60b902b767bc3ccd97fb49bb3b55580 (diff) | |
| parent | 86f95070c0d3d00ebe133caa823fdb26e28baa59 (diff) | |
Merge pull request #22 from rowjay/initializer
update deprecated tf.initialize_all_variables to tf.global_variables_…
| -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 |