transformers
816f4424
- Only cast logits to float when computing loss (#34147)
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
1 year ago
Only cast logits to float when computing loss (#34147) * Only cast logits to float when computing loss Some misses from #31292 and #33902 * Move logits.float() into existing if labels is not None branch
References
#34147 - Only cast logits to float when computing loss
Author
Matthew Hoffman
Parents
e46e3bc1
Loading