Lightning Integration
tchaton opened this issue · comments
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