chenyh
/
MNIST_Example
mirror of https://openi.pcl.ac.cn/OpenIOSSG/MNIST_Example.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Merge branch 'develop' of https://git.openi.org.cn/OpenIOSSG/MNIST_Example into develop

This commit is contained in:
liuzx 2022-08-10 10:04:28 +08:00
parent 7abcf25499 cf40a66c9c
commit 8e261a2c00
1 changed files with 55 additions and 55 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
dataset_distributed.py
View File

@ -14,7 +14,7 @@ from mindspore.dataset.vision import Inter
from mindspore.common import dtype as mstype from mindspore.common import dtype as mstype
from mindspore.communication.management import init, get_rank, get_group_size from mindspore.communication.management import init, get_rank, get_group_size
init()
def create_dataset_parallel(data_path, batch_size=32, repeat_size=1, def create_dataset_parallel(data_path, batch_size=32, repeat_size=1,
num_parallel_workers=1, shard_id=0, num_shards=8): num_parallel_workers=1, shard_id=0, num_shards=8):
""" """