Hey @FruitVinegar , I think this time your question will also help other users:
I plan to create an example of QAT using the code you shared here to train a resnet18 on CIFAR10.
If you’d like to take a look and comment out, it could be helpful : Added Cifar10 QAT example by vfdev-5 · Pull Request #1556 · pytorch/ignite · GitHub 4
Thanks !