Returns a tensorboard summary writer
(name, base="..", iteration=0)
| 31 | |
| 32 | |
| 33 | def get_sample_writer(name, base="..", iteration=0): |
| 34 | """Returns a tensorboard summary writer |
| 35 | """ |
| 36 | return SummaryWriter( |
| 37 | log_dir=os.path.join(base, SUMMARY_WRITER_DIR_NAME, name), purge_step=iteration) |
| 38 | |
| 39 | |
| 40 | def print_rank_0(message): |