Advertisement
Guest User

Untitled

a guest
Jun 19th, 2019
88
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.52 KB | None | 0 0
  1. var_list = tf.contrib.framework.get_variables_to_restore(exclude=new_variables_name)
  2. saver = tf.train.Saver(var_list)
  3.  
  4. with tf.train.MonitoredTrainingSession(checkpoint_dir=FLAGS.train_dir,
  5. hooks=[tf.train.StopAtStepHook(last_step=FLAGS.max_steps),
  6. tf.train.NanTensorHook(loss)],
  7. config=tf.ConfigProto(
  8. log_device_placement=FLAGS.log_device_placement)) as mon_sess:
  9. saver.restore(mon_sess, ".../model.ckpt-100000")
  10. while not mon_sess.should_stop():
  11. mon_sess.run(train_op)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement