Only cast logits to float when computing loss #34147
ArthurZucker
merged
2 commits into
huggingface:main
from
ringohoffman:only-cast-logits-float-compute-loss
Only cast logits to float when computing loss
2eaa05e3
gante
approved these changes
on 2024-10-17
Move logits.float() into existing if labels is not None branch
b9209618
ArthurZucker
approved these changes
on 2024-10-18
Login to write a write a comment.
Login via GitHub