Do you have saved models available?

[admk]

Could you provide us with saved models?

Thanks!

[htwang14]

Our pretrained ATMC 8bit models are available at https://drive.google.com/drive/folders/1HDGkz8BKQGn6SFRPzaTt13hFOC8JbtUD?usp=sharing

The files are names as quant_closest_<dataset>_atmc_<cr>.pth. For example, the file quant_closest_svhn_atmc_0.005.pth is the ATMC 8-bit model on SVHN dataset with 0.005 compression ratio (0.5% non-zero model parameters).

You can evaluated them using {dataset}/inference_model.py. For example, use the following command to evaluate robustness against PGD attacks with eps=4 (controlled by attack_eps parameter) and iter=7 (controlled by attack_iter parameter) on SVHN dataset. (Set loaddir and data_root as proper file paths on your own machine.)

python svhn/inference_model.py --gpu 0 --loaddir /ckpt/root/path/svhn/atmc_8 --model_name svhn_atmc_0.1.pth --prefix_name quant_closest_ --data_root /data/root/path/svhn --attack_algo pgd --attack_eps 4 --attack_iter 7 --prune_algo l0proj --prune_ratio 0.1 --quantize_algo kmeans_nnz_fixed_0_center --quantize_bits 8 --exp_logger svhn_atmc_8.txt --abc_special --abc_initialize

[admk]

Sorry my question wasn't specific enough, do you have CIFAR-10 & MNIST models? Thank you!