!1291 add import ckpt

Merge pull request !1291 from meixiaowei/master
pull/1291/MERGE
mindspore-ci-bot 5 years ago committed by Gitee
commit 979d2e23dc

@ -87,7 +87,7 @@ sh run_distribute_train.sh rank_table_8p.json dataset/ilsvrc ./ckpt/pretrained.c
# standalone training example1p
sh run_standalone_train.sh dataset/ilsvrc
f you want to load pretrained ckpt file,
If you want to load pretrained ckpt file,
sh run_standalone_train.sh dataset/ilsvrc ./ckpt/pretrained.ckpt
```

@ -28,6 +28,7 @@ from mindspore.nn.optim.momentum import Momentum
from mindspore.train.model import Model, ParallelMode
from mindspore.train.callback import ModelCheckpoint, CheckpointConfig, LossMonitor, TimeMonitor
from mindspore.train.loss_scale_manager import FixedLossScaleManager
from mindspore.train.serialization import load_checkpoint, load_param_into_net
import mindspore.dataset.engine as de
from mindspore.communication.management import init
import mindspore.nn as nn

Loading…
Cancel
Save