Skip to content

Commit

Permalink
set default of 'warming_up' to 0
Browse files Browse the repository at this point in the history
  • Loading branch information
TsumiNa committed Jun 1, 2021
1 parent a50301c commit fd3857c
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions xenonpy/model/training/extension/validator.py
Original file line number Diff line number Diff line change
Expand Up @@ -27,7 +27,7 @@ def __init__(self,
each_iteration: bool = True,
early_stopping: int = None,
trace_order: int = 1,
warming_up: int = 50,
warming_up: int = 0,
**trace_criteria: Dict[str, float]):
"""
Expand Down Expand Up @@ -121,7 +121,7 @@ def _validate():
for name, (target, current) in self.trace.items():
if name in metrics:
score = np.abs(metrics[name] - target)
if score < current[-1] and step_info['i_epoch'] > self._warming_up:
if score < current[-1] and step_info['i_epoch'] >= self._warming_up:
current.append(score)
current.sort()
current.pop()
Expand Down

0 comments on commit fd3857c

Please sign in to comment.