Browse Source

fix getdatasetsize error II

tags/v0.7.0-beta
panfengfeng 5 years ago
parent
commit
57eab288cd
1 changed files with 2 additions and 1 deletions
  1. +2
    -1
      mindspore/train/model.py

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

@@ -16,6 +16,7 @@
from collections.abc import Iterable from collections.abc import Iterable


import os import os
import math
import numpy as np import numpy as np


from mindspore import log as logger from mindspore import log as logger
@@ -402,7 +403,7 @@ class Model:
if sink_size == -1: if sink_size == -1:
epoch_num = epoch epoch_num = epoch
else: 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, dataset_helper, train_network = self._exec_preprocess(self._train_network,
is_train=True, is_train=True,


Loading…
Cancel
Save