transformers
d26c1413 - [SegGPT] Fix loss calculation (#30421)
Commit
View On GitHub
Commit
1 year ago
[SegGPT] Fix loss calculation (#30421) * Fixed main train issues * Added loss test * Update src/transformers/models/seggpt/modeling_seggpt.py Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com> * Added missing labels arg in SegGptModel forward * Fixed typo * Added slow test to test loss calculation --------- Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com>
Author
EduardoPach EduardoPach
Parents
37fa1f65
FAQ Terms Privacy Refunds Impressum
Loading