Mixed precision training
hemangchawla opened this issue · 0 comments
hemangchawla commented
Hello,
Do you have an example on how to use mixed precision training for smaller GPU memory?
I see that apex is part of the Dockerfile. However, I am not sure where the amp is used.
Could you please assist?
Kind regards