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

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,


Loading…
Cancel
Save