Skip to content
Snippets Groups Projects
Unverified Commit 03294cfb authored by i-robot's avatar i-robot Committed by Gitee
Browse files

!3030 fix train for scop 

Merge pull request !3030 from 185******25/master
parents 7538e31b d8f08592
No related branches found
No related tags found
No related merge requests found
Hide whitespace changes
Inline Side-by-side
official/cv/resnet/golden_stick/pruner/scop/train.py
+ 2
− 1
View file @ 03294cfb
......@@ -25,7 +25,8 @@ from mindspore.communication.management import init, get_rank
from mindspore.parallel import set_algo_parameters
from mindspore.train.loss_scale_manager import FixedLossScaleManager
from mindspore.nn.loss import SoftmaxCrossEntropyWithLogits
from mindspore_gs import PrunerKfCompressAlgo, PrunerFtCompressAlgo, KfConv2d, MaskedConv2dbn
from mindspore_gs import PrunerKfCompressAlgo, PrunerFtCompressAlgo
from mindspore_gs.pruner.scop.scop_pruner import KfConv2d, MaskedConv2dbn
from src.lr_generator import get_lr
from src.CrossEntropySmooth import CrossEntropySmooth
from src.resnet import conv_variance_scaling_initializer
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment