diff --git a/official/cv/resnet/golden_stick/pruner/scop/train.py b/official/cv/resnet/golden_stick/pruner/scop/train.py index 9db5ed87d06873ee1cd539aed43cee5a43364156..107189623c122932bda92745bc968a45b143b6d3 100644 --- a/official/cv/resnet/golden_stick/pruner/scop/train.py +++ b/official/cv/resnet/golden_stick/pruner/scop/train.py @@ -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