├── INSTALL.md ├── README.md ├── multi_quantization ├── __init__.py ├── checkpoint.py ├── prediction.py └── quantization.py ├── requirements.txt ├── setup.py └── tests ├── test_quantization.py ├── test_train_hdf5.py └── test_write_hdf5.py /INSTALL.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/k2-fsa/multi_quantization/HEAD/INSTALL.md -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/k2-fsa/multi_quantization/HEAD/README.md -------------------------------------------------------------------------------- /multi_quantization/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/k2-fsa/multi_quantization/HEAD/multi_quantization/__init__.py -------------------------------------------------------------------------------- /multi_quantization/checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/k2-fsa/multi_quantization/HEAD/multi_quantization/checkpoint.py -------------------------------------------------------------------------------- /multi_quantization/prediction.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/k2-fsa/multi_quantization/HEAD/multi_quantization/prediction.py -------------------------------------------------------------------------------- /multi_quantization/quantization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/k2-fsa/multi_quantization/HEAD/multi_quantization/quantization.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | torch 2 | numpy 3 | h5py 4 | -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/k2-fsa/multi_quantization/HEAD/setup.py -------------------------------------------------------------------------------- /tests/test_quantization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/k2-fsa/multi_quantization/HEAD/tests/test_quantization.py -------------------------------------------------------------------------------- /tests/test_train_hdf5.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/k2-fsa/multi_quantization/HEAD/tests/test_train_hdf5.py -------------------------------------------------------------------------------- /tests/test_write_hdf5.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/k2-fsa/multi_quantization/HEAD/tests/test_write_hdf5.py --------------------------------------------------------------------------------