增加多卡数据并行训练示例
This commit is contained in:
parent
79010c9e26
commit
142411e353
|
@ -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
|
||||
|
|
Loading…
Reference in New Issue