Showing
2 changed files
with
4 additions
and
4 deletions
... | @@ -190,7 +190,7 @@ def load_model(model, input_map=None): | ... | @@ -190,7 +190,7 @@ def load_model(model, input_map=None): |
190 | print('Checkpoint file: %s' % ckpt_file) | 190 | print('Checkpoint file: %s' % ckpt_file) |
191 | 191 | ||
192 | saver = tf.compat.v1.train.import_meta_graph(os.path.join(model_exp, meta_file), input_map=input_map) | 192 | saver = tf.compat.v1.train.import_meta_graph(os.path.join(model_exp, meta_file), input_map=input_map) |
193 | - saver.restore(tf.get_default_session(), os.path.join(model_exp, ckpt_file)) | 193 | + saver.restore(tf.compat.v1.get_default_session(), os.path.join(model_exp, ckpt_file)) |
194 | 194 | ||
195 | def get_model_filenames(model_dir): | 195 | def get_model_filenames(model_dir): |
196 | files = os.listdir(model_dir) | 196 | files = os.listdir(model_dir) | ... | ... |
-
Please register or login to post a comment