Skip to content

Commit

Permalink
fix argument for early_stop (#3573)
Browse files Browse the repository at this point in the history
  • Loading branch information
ooooo-create committed Nov 15, 2023
1 parent e167b03 commit 0007174
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions tools/train.py
Original file line number Diff line number Diff line change
Expand Up @@ -85,8 +85,8 @@ def parse_args():
parser.add_argument(
'--early_stop_intervals',
help='Early Stop at args number of save intervals.',
type=None,
default=0)
type=int,
default=None)

# Other params
parser.add_argument(
Expand Down

0 comments on commit 0007174

Please sign in to comment.