Commit 1775c931 authored by Julian Zimmermann's avatar Julian Zimmermann
Browse files

del bids assertion

parent 83509ce6
......@@ -356,7 +356,7 @@ def perform_evaluation(model, eval_steps, ckpt, strategy):
pe_list.extend(pe.numpy())
add_list.extend(add.numpy())
logging.info('Completed eval for %d / %d steps', i + 1, eval_steps)
assert len(unique(bi_list)) == len(bi_list)
# assert len(unique(bi_list)) == len(bi_list)
savez(os.path.join(FLAGS.model_dir, "final_projections_epoch-{}.npz".format(FLAGS.train_epochs)),
labels=label_list, projections=projections_list, images=images_list,
bunch_ids=bi_list, photon_energies=pe_list, additional=add_list, images_t=images_t_list)
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment