Create a tensorboard that saves losses.
(tensorboard_dir)
| 202 | |
| 203 | |
| 204 | def create_tensorboard(tensorboard_dir): |
| 205 | """ |
| 206 | Create a tensorboard that saves losses. |
| 207 | """ |
| 208 | if dist.get_rank() == 0: # real tensorboard |
| 209 | # tensorboard |
| 210 | writer = SummaryWriter(tensorboard_dir) |
| 211 | |
| 212 | return writer |
| 213 | |
| 214 | def write_tensorboard(writer, *args): |
| 215 | ''' |
nothing calls this directly
no outgoing calls
no test coverage detected