search

Zhen-Dong / HAWQ

Quantization library for PyTorch. Support low-precision and mixed-precision quantization, with hardware implementation through TVM.
MIT License
410 stars 83 forks source link

No module named 'bit_config' #1

Closed leiwen83 closed 3 years ago

leiwen83 commented 3 years ago

Hi,

I try to run quant_train.py, but meet the error as below, how to solve it ?

Traceback (most recent call last): File "quant_train.py", line 22, in from bit_config import * ModuleNotFoundError: No module named 'bit_config'

Thx, Lei

Zhen-Dong commented 3 years ago

Hi Lei, thanks for pointing this out. I think this issue was solved by some recent commits, so please pull the current repo. Specifically, we moved the bit_config.py into the tvm_benchmark folder and imported the module from there. Feel free to reach out if this doesn't work.

Zhen-Dong commented 3 years ago

To follow up, I submit a new commit moving bit_config.py out of the tvm_benchmark folder, I think this could make the codes more straightforward.

leiwen83 commented 3 years ago

great! it works!