Returns a tensorboard summary writer
(log_dir, iteration=0)
| 35 | |
| 36 | |
| 37 | def get_sample_writer(log_dir, iteration=0): |
| 38 | """Returns a tensorboard summary writer |
| 39 | """ |
| 40 | return SummaryWriter( |
| 41 | log_dir=log_dir, purge_step=iteration) |
| 42 | |
| 43 | |
| 44 | def print_rank_0(message): |