This website works better with JavaScript.
Home
Issues
Pull Requests
Milestones
AI流水线
Repositories
Datasets
Forum
实训
竞赛
大数据
Register
Sign In
hummingbird
/
mindarmour
Not watched
Unwatch
Watch all
Watch but not notify
2
Star
0
Fork
0
Code
Releases
17
Wiki
Activity
Issues
0
Pull Requests
0
Datasets
Model
Cloudbrain
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
2e1e351473
c214f550f0
commit
16b4bcbcdb
1 changed files
with
1 additions
and
2 deletions
Split View
Diff Options
Show Stats
Download Patch File
Download Diff File
+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")
Write
Preview
Loading…
Cancel
Save