Dongyue Wu
Dongyue Wu
So far, following the provided code in [benchmark/readme.md](https://github.com/VainF/Torch-Pruning/blob/master/benchmarks/readme.md), I only get **75.32**. I set the batchsize=128 and using 2 GPUs for training with lr=0.01 and method=l1. The rest of settings...
Here's the script: `CUDA_VISIBLE_DEVICES=0,1 python main.py --prune_method opp --opp_scheme v5 --lw_opp 1000 --update_reg_interval 5 --stabilize_reg_interval 40000 --dataset imagenet -a resnet50 --pretrained --lr_ft 0:0.01,30:0.001,60:0.0001,75:0.00001 --epochs 90 --batch_size_prune 256 --batch_size 256 --index_layer...