Skip to content

Commit

Permalink
[FIx] Fix the total number of iterations in log is a float number. (#604
Browse files Browse the repository at this point in the history
)

* Update

* Ensure max_iters being integer number
  • Loading branch information
jbwang1997 committed Oct 18, 2022
1 parent 51ac3a2 commit 4111cfb
Showing 1 changed file with 7 additions and 3 deletions.
10 changes: 7 additions & 3 deletions mmengine/runner/loops.py
Original file line number Diff line number Diff line change
Expand Up @@ -41,8 +41,10 @@ def __init__(
val_interval: int = 1,
dynamic_intervals: Optional[List[Tuple[int, int]]] = None) -> None:
super().__init__(runner, dataloader)
self._max_epochs = max_epochs
self._max_iters = max_epochs * len(self.dataloader)
self._max_epochs = int(max_epochs)
assert self._max_epochs == max_epochs, \
f'`max_epochs` should be a integer number, but get {max_epochs}.'
self._max_iters = self._max_epochs * len(self.dataloader)
self._epoch = 0
self._iter = 0
self.val_begin = val_begin
Expand Down Expand Up @@ -206,7 +208,9 @@ def __init__(
val_interval: int = 1000,
dynamic_intervals: Optional[List[Tuple[int, int]]] = None) -> None:
super().__init__(runner, dataloader)
self._max_iters = max_iters
self._max_iters = int(max_iters)
assert self._max_iters == max_iters, \
f'`max_iters` should be a integer number, but get {max_iters}'
self._max_epochs = 1 # for compatibility with EpochBasedTrainLoop
self._epoch = 0
self._iter = 0
Expand Down

0 comments on commit 4111cfb

Please sign in to comment.