[Bugbash] promote pruning v2 (#4733)

examples/model_compress/pruning/basic_pruners_torch.py → examples/model_compress/pruning/legacy/basic_pruners_torch.py

@@ -18,12 +18,12 @@ from torch.optim.lr_scheduler import StepLR, MultiStepLR
 from torchvision import datasets, transforms
 
 from pathlib import Path
-sys.path.append(str(Path(__file__).absolute().parents[1] / 'models'))
+sys.path.append(str(Path(__file__).absolute().parents[2] / 'models'))
 from mnist.lenet import LeNet
 from cifar10.vgg import VGG
 from cifar10.resnet import ResNet18
 
-from nni.compression.pytorch.utils.counter import count_flops_params
+from nni.compression.pytorch.utils import count_flops_params
 
 import nni
 from nni.compression.pytorch import ModelSpeedup

examples/model_compress/pruning/finetune_kd_torch.py → examples/model_compress/pruning/legacy/finetune_kd_torch.py

@@ -19,7 +19,7 @@ from torch.optim.lr_scheduler import MultiStepLR
 from basic_pruners_torch import get_data
 
 from pathlib import Path
-sys.path.append(str(Path(__file__).absolute().parents[1] / 'models'))
+sys.path.append(str(Path(__file__).absolute().parents[2] / 'models'))
 from mnist.lenet import LeNet
 from cifar10.vgg import VGG
 

examples/model_compress/pruning/mobilenetv2_end2end/Compressing MobileNetV2 with NNI Pruners.ipynb → examples/model_compress/pruning/legacy/mobilenetv2_end2end/Compressing MobileNetV2 with NNI Pruners.ipynb

@@ -34,7 +34,7 @@
     "import numpy as np\n",
     "\n",
     "from nni.compression.pytorch import ModelSpeedup\n",
-    "from nni.compression.pytorch.utils.counter import count_flops_params\n",
+    "from nni.compression.pytorch.utils import count_flops_params\n",
     "\n",
     "from utils import create_model, get_dataloader\n",
     "\n",