search

VITA-Group / ATMC

[NeurIPS'2019] Shupeng Gui, Haotao Wang, Haichuan Yang, Chen Yu, Zhangyang Wang, Ji Liu, “Model Compression with Adversarial Robustness: A Unified Optimization Framework”
MIT License
49 stars 11 forks source link

Uncompleted code #1

Closed linlin123git closed 4 years ago

linlin123git commented 4 years ago

Hi,

When I tried to run the codes, some modules seem to be uncompleted yet.

python mnist/train_proj_admm_quant.py

Error message: from super_module.super_class import Linear_QuantForward, Conv2d_QuantForward ImportError: cannot import name 'Linear_QuantForward'

ImportError: No module named 'util_unnormalize'

I searched the repository but cannot find these modules. Can you please check it?

Many thanks.

shupenggui commented 4 years ago

Hi,

Thanks for pointing out the problem. We have cleaned the repo again and remove the dependency on Linear_QuantForward, Conv2d_QuantForward. Please try to pull and run it again.