增加多卡数据并行训练示例

This commit is contained in:
liuzx 2022-07-19 10:37:15 +08:00
parent 79010c9e26
commit 142411e353
1 changed files with 2 additions and 1 deletions

View File

@ -73,6 +73,7 @@ from dataset import create_dataset
from lenet import LeNet5
import mindspore.nn as nn
from mindspore import context
from mindspore.common import set_seed
from mindspore.train.callback import ModelCheckpoint, CheckpointConfig, LossMonitor, TimeMonitor
from mindspore.train import Model
from mindspore.nn.metrics import Accuracy
@ -137,7 +138,7 @@ parser.add_argument('--epoch_size',
type=int,
default=5,
help='Training epochs.')
set_seed(114514)
if __name__ == "__main__":
args = parser.parse_args()
### defining the training environment