Skip to content

Commit

Permalink
Fixup lint
Browse files Browse the repository at this point in the history
  • Loading branch information
erogol committed Dec 12, 2023
1 parent a5f9f17 commit 0741bbd
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion trainer/io.py
Original file line number Diff line number Diff line change
Expand Up @@ -188,7 +188,7 @@ def save_best_model(
else:
is_save_model = current_loss < best_loss

if isinstance(keep_after, int) or isinstance(keep_after, float):
if isinstance(keep_after, (int, float)):
keep_after = int(keep_after)
is_save_model = is_save_model and current_step > keep_after

Expand Down

0 comments on commit 0741bbd

Please sign in to comment.