Skip to content
Snippets Groups Projects
Unverified Commit d8f08592 authored by 185******25's avatar 185******25 Committed by Gitee
Browse files

update official/cv/resnet/golden_stick/pruner/scop/train.py.

parent 7538e31b
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 @ d8f08592
......@@ -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