(self)
| 41 | self.tag_mode_exceptions = {'add_histogram', 'add_embedding'} # TODO : Test these two funcs. |
| 42 | |
| 43 | def dump_stats(self): |
| 44 | with open(f"{self.log_dir}/log", "w") as f: |
| 45 | json.dump(self.stats, f, |
| 46 | indent=4, |
| 47 | ensure_ascii=False, |
| 48 | separators=(",", ": "), |
| 49 | ) |
| 50 | |
| 51 | def set_epoch(self, epoch, mode): |
| 52 | ''' |