- Pytorch 1.3+
- Python 3+
- CUDA 10+
├── quantrans
│ ├── BaseQuanTransformer.py
│ ├── builder.py
│ ├── __init__.py
│ ├── quantops
│ ├── QuanTransformerV1.py
│ ├── QuanTransformerV2.py
│ └── utils
├── README.md
└── setup.py
Step 1: Install QuantQuant
git clone https://github.com/ProhetTeam/QuantQuant.git
cd QuantQuant
pip3 install -r requirements.txt
Step 2 : clone the submoddule
git submodule init
git submodule update
#or git submodule update --init --recursive
Step 3: Install quantrans & quantops
cd QuanTransformer
python3 setup.py install # or pip3 install -v -e .
From then, you can use the different quant operators in your projects.
To verify that you can use the quant operators normally, you can test in terminal as follows:
>>>python
>>>from quantrans.quantops import LSQ
>>>
Also, you can use the QuanTransformer()
function from quantrans
.