(saver, sess, logdir, step)
| 107 | |
| 108 | |
| 109 | def save(saver, sess, logdir, step): |
| 110 | model_name = 'model.ckpt' |
| 111 | checkpoint_path = os.path.join(logdir, model_name) |
| 112 | print('Storing checkpoint to {} ...'.format(logdir), end="") |
| 113 | sys.stdout.flush() |
| 114 | |
| 115 | if not os.path.exists(logdir): |
| 116 | os.makedirs(logdir) |
| 117 | |
| 118 | saver.save(sess, checkpoint_path, global_step=step) |
| 119 | print(' Done.') |
| 120 | |
| 121 | |
| 122 | def load(saver, sess, logdir): |