forked from mindspore-Ecosystem/mindspore
repair unsuitable warning in repeat count and epoch size check
This commit is contained in:
parent
dcd1f0a504
commit
59d08edea7
mindspore/train
|
@ -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…
Reference in New Issue