lsq-net
lsq-net copied to clipboard
Unofficial implementation of LSQ-Net, a neural network quantization framework
Thanks for your the implementation! In LsqQuan. https://github.com/zhutmost/lsq-net/blob/2c24a96be06d044fa4c7d651727f4574b8d88c86/quan/quantizer/lsq.py#L54 If s_scale contains 0, there will be Divided by 0 Error, and the training loss will become nan. Maybe eps should be...
I add distillation when training resnet18. But the Top-1 Acc degrades from 68.150 % to 67.364%。 Hyperparameters as follow: 4gpu epochs: 90 learning_rate: 0.01 momentum: 0.9 weight_decay: 0.0001 mode: step...
python main.py config.yaml INFO - Log file for this run: /home/itl/Documents/xrh/lsq-net/out/MyProject_20220505-162725/MyProject_20220505-162725.log INFO - TensorBoard data directory: /home/itl/Documents/xrh/lsq-net/out/MyProject_20220505-162725/tb_runs Files already downloaded and verified Files already downloaded and verified INFO - Dataset...
Hi, first, thanks for your implementation! It's not too hard to apply your code to my current model! However, the dumped checkpoint has the same size as the original model,...
When we have trained the quantization model, how to deploy it in cpu backend?
Hey, thanks for your work. Have you tested the accuracy for Cifar10? I tried to quantized ResNet20 into 2/2 4/4 bit, and the accuracy only 88%~90%. If possible, can you...
Hi, Great implementation! Since per-channel weight quantization is implemented in you code, I'm wondering if there is any improvement compared to per-tensor weight quantization.
Thanks for your work! I have 2 questions: In the paper, the author use cos lr but why fixsLR used in your code? And now W3A2 acc has reported but...
Thanks for your great work! But I trained resnet18 with the default config, and only got the top1 acc 52, top5 acc 75
When using the `random_split` function provided in PyTorch to split the dataset to training and validation sets, the propotion of categories are imbalanced.