Quantcast
Channel: LightningModule - Lightning AI
Viewing all articles
Browse latest Browse all 34

Disabling autocast for certain modules

$
0
0

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 34

Trending Articles