Skip to content

Commit 563d4dd

Browse files
committed
Avoid division by zero error
1 parent f9cc267 commit 563d4dd

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

machine/translation/huggingface/hugging_face_nmt_model_trainer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -384,7 +384,7 @@ def __init__(
384384
self._max_steps = max_steps if max_steps is not None else 0
385385
self._progress = progress
386386
self._check_canceled = check_canceled
387-
self._update_frequency = update_frequency if update_frequency is not None else (self._max_steps // 100)
387+
self._update_frequency = update_frequency if update_frequency is not None else max((self._max_steps // 100), 1)
388388

389389
def on_train_begin(self, args: TrainingArguments, state: TrainerState, control: TrainerControl, **kwargs) -> None:
390390
if self._check_canceled is not None:

0 commit comments

Comments
 (0)