yhhhli / APoT_Quantization

PyTorch implementation for the APoT quantization (ICLR 2020)
265 stars 51 forks source link

Lightning Integration #13

Open tchaton opened 3 years ago

tchaton commented 3 years ago

Dear @yhhhli,

This work looks awesome ! I was wondering if you would be interested in making a PR contribution to PyTorch Lightning. It could a callback converting users model and apply this quantisation method for training the model. It would fit really nicely with pruning, fine-tuning.

Best, T.C