Skip to content
Snippets Groups Projects
Commit 7d1d308f authored by tomrink's avatar tomrink
Browse files

snapshot...

parent a9fda5b2
Branches
No related tags found
No related merge requests found
......@@ -772,7 +772,7 @@ def run_restore_static(directory, ckpt_dir, out_file=None):
labels, preds, inputs = nn.run_restore(directory, ckpt_dir)
if out_file is not None:
np.save(out_file,
[np.squeeze(labels), preds.argmax(axis=3), np.squeeze(inputs)])
[np.squeeze(labels), preds.argmax(axis=3), inputs])
def run_evaluate_static(in_file, out_file, ckpt_dir):
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment