Quantcast
Viewing all articles
Browse latest Browse all 36

Disabling autocast for certain modules

Hi,
I was wondering what is the way in Lightning to disable mixed precision for certain sub-modules?
Is there a way to do this through callbacks?

Thanks

2 posts - 2 participants

Read full topic


Viewing all articles
Browse latest Browse all 36

Trending Articles