Nasnet pruning doesn't work

VainF / Torch-Pruning

[CVPR 2023] Towards Any Structural Pruning; LLMs / SAM / Diffusion / Transformers / YOLOv8 / CNNs

https://arxiv.org/abs/2301.12900

MIT License

2.6k stars 321 forks source link

Nasnet pruning doesn't work #127

Open johnrachwan123 opened 1 year ago

johnrachwan123 commented 1 year ago

Hey, simply running: python main.py --mode prune --model nasnet --batch-size 128 --restore True --dataset cifar100 --method random --speed-up 10 --global-pruning

Doesn't work to prune Nasnet. Is this an architecture you have already tested ?

VainF commented 1 year ago

Hello, @johnrachwan123. Currently, we have not conducted any testing of NASNet on CIFAR. However, our pruner has been verified to function properly on MNASNet from Torchvision. I will double-check this information and get back to you.