liufinback

> pruning code: > > ``` > from nni.compression.pytorch.speedup.v2 import ModelSpeedup > pruner = L1NormPruner(model, config_list) > _, masks = pruner.compress() > pruner._unwrap_model() > ModelSpeedup(model, torch.randn(1, 3, 1024, 1024).to(device), masks_or_file=masks).speedup_model()...

If anyone can help. Thanks a lot. Please also send on copy to my email [email protected]