p04896132
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

add import ckpt

This commit is contained in:
meixiaowei 2020-05-20 16:15:19 +08:00
parent 44e3d167f0
commit 863513e8dc
2 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
example/resnet101_imagenet2012/README.md
View File

@ -87,7 +87,7 @@ sh run_distribute_train.sh rank_table_8p.json dataset/ilsvrc ./ckpt/pretrained.c
# standalone training example（1p）
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
```

1
example/resnet101_imagenet2012/train.py
View File

@ -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