提交 9fff0508 编写于 作者: M mindspore-ci-bot 提交者: Gitee

!3469 fix getdataset size error II

Merge pull request !3469 from panfengfeng/fix_getdataset_size_error_II
......@@ -16,6 +16,7 @@
from collections.abc import Iterable
import os
import math
import numpy as np
from mindspore import log as logger
......@@ -402,7 +403,7 @@ class Model:
if sink_size == -1:
epoch_num = epoch
else:
epoch_num = epoch * sink_size // train_dataset.get_dataset_size()
epoch_num = math.ceil(epoch * sink_size / train_dataset.get_dataset_size())
dataset_helper, train_network = self._exec_preprocess(self._train_network,
is_train=True,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册