please check this GitHub repo:
Hi, congratulations and thanks for sharing.
I was wondering what you thought of torch.cuda.amp. I am asking beacuse (it's new and) I haven't had time to try it out myself.
Hi, thank you.
the torch.cuda.amp is used to benefit from the mixed-precision approach which makes training faster.