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

2022-08-10 10:04:28 +08:00 · 2022-08-10 10:04:28 +08:00 · 8e261a2c00
parent 7abcf25499 cf40a66c9c
commit 8e261a2c00
1 changed files with 55 additions and 55 deletions
--- a/dataset_distributed.py
+++ b/dataset_distributed.py
@ -14,7 +14,7 @@ from mindspore.dataset.vision import Inter
 from mindspore.common import dtype as mstype
 from mindspore.communication.management import init, get_rank, get_group_size

-
+init()
 def create_dataset_parallel(data_path, batch_size=32, repeat_size=1,
                   num_parallel_workers=1, shard_id=0, num_shards=8):
    """