Browse Source

!58 Optimize GetDatasize

Merge pull request !58 from jiangzhiwen/get_data_size
tags/v0.6.0-beta
mindspore-ci-bot Gitee 5 years ago
parent
commit
16b4bcbcdb
1 changed files with 1 additions and 2 deletions
  1. +1
    -2
      example/mnist_demo/lenet5_dp_pynative_model.py

+ 1
- 2
example/mnist_demo/lenet5_dp_pynative_model.py View File

@@ -99,8 +99,7 @@ if __name__ == "__main__":

# get training dataset
ds_train = generate_mnist_dataset(os.path.join(cfg.data_path, "train"),
cfg.batch_size,
cfg.epoch_size)
cfg.batch_size)

if cfg.micro_batches and cfg.batch_size % cfg.micro_batches != 0:
raise ValueError("Number of micro_batches should divide evenly batch_size")


Loading…
Cancel
Save