(self, tb_dir, file_name, use_tensorboard=False)
| 137 | """ |
| 138 | |
| 139 | def __init__(self, tb_dir, file_name, use_tensorboard=False): |
| 140 | self.tb_dir = tb_dir |
| 141 | self.use_tensorboard = use_tensorboard |
| 142 | if self.use_tensorboard: |
| 143 | self.writer = SummaryWriter(os.path.join(self.tb_dir, file_name)) |
| 144 | else: |
| 145 | self.writer = CustomWriter(os.path.join(self.tb_dir, file_name)) |
| 146 | |
| 147 | def update(self, tb_dict, it, suffix=None, mode="train"): |
| 148 | """ |
nothing calls this directly
no test coverage detected