Skip to content
Snippets Groups Projects
Unverified Commit eb251299 authored by Mashiro's avatar Mashiro Committed by GitHub
Browse files

[Feature]: add epoch log scalar (#391)

parent f850de71
No related branches found
No related tags found
No related merge requests found
...@@ -143,6 +143,7 @@ class LogProcessor: ...@@ -143,6 +143,7 @@ class LogProcessor:
if self.by_epoch: if self.by_epoch:
if mode in ['train', 'val']: if mode in ['train', 'val']:
cur_epoch = self._get_epoch(runner, mode) cur_epoch = self._get_epoch(runner, mode)
tag['epoch'] = cur_epoch
log_str = (f'Epoch({mode}) [{cur_epoch}]' log_str = (f'Epoch({mode}) [{cur_epoch}]'
f'[{cur_iter}/{len(current_loop.dataloader)}] ') f'[{cur_iter}/{len(current_loop.dataloader)}] ')
else: else:
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment