Browse Source

repair unsuitable warning in repeat count and epoch size check

tags/v0.3.0-alpha
ms_yan chang zherui 5 years ago
parent
commit
59d08edea7
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/train/model.py

+ 1
- 1
mindspore/train/model.py View File

@@ -372,7 +372,7 @@ class Model:
>>> model.train(2, dataset)
"""
repeat_count = train_dataset.get_repeat_count()
if epoch != repeat_count:
if epoch != repeat_count and dataset_sink_mode is True:
logger.warning(f"The epoch_size {epoch} is not the same with dataset repeat_count {repeat_count}")
check_bool(dataset_sink_mode)
_device_number_check(self._parallel_mode, self._device_number)


Loading…
Cancel
Save