From 863513e8dc0daceba7706c7bbc17d7eabf2b89a7 Mon Sep 17 00:00:00 2001 From: meixiaowei Date: Wed, 20 May 2020 16:15:19 +0800 Subject: [PATCH] add import ckpt --- example/resnet101_imagenet2012/README.md | 2 +- example/resnet101_imagenet2012/train.py | 1 + 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/example/resnet101_imagenet2012/README.md b/example/resnet101_imagenet2012/README.md index 6578b09f0ec..6ccaf5f6b6e 100644 --- a/example/resnet101_imagenet2012/README.md +++ b/example/resnet101_imagenet2012/README.md @@ -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 ``` diff --git a/example/resnet101_imagenet2012/train.py b/example/resnet101_imagenet2012/train.py index 06104941e9d..e3d6adb267e 100755 --- a/example/resnet101_imagenet2012/train.py +++ b/example/resnet101_imagenet2012/train.py @@ -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