Quantize Aware training on Style GAN 3

While going through the styleGAN3 code, I did observe that multi precision is supported in the form of fp16 and fp 32. However, I wanted to know if StyleGAN 3 would support Pytorch’s Quantize aware training?
Any thoughts on that would be greatly appreciated

Hi @user16955 ,
This doesn’t look like a cudnn question.
You may get better assistance for the same in Pytorch Forum.
Thanks!