I’m using pytorch 1.7.1 now.
Trying to use QAT for some project.
Everything works well on training.
But when I tried to inference on GPU,
Some errors appeared.
Is QAT inference not support GPU at this moment?
Will it be supported at later patch?