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

[Fix] Fix LogProcessor._get_iter (#370)

parent 1241c212
No related branches found
No related tags found
No related merge requests found
...@@ -369,7 +369,7 @@ class LogProcessor: ...@@ -369,7 +369,7 @@ class LogProcessor:
Returns: Returns:
int: The current global iter or inner iter. int: The current global iter or inner iter.
""" """
if self.by_epoch and batch_idx: if self.by_epoch and batch_idx is not None:
current_iter = batch_idx + 1 current_iter = batch_idx + 1
else: else:
current_iter = runner.iter + 1 current_iter = runner.iter + 1
......
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