PR #30421 [SegGPT] Fix loss calculation

[SegGPT] Fix loss calculation #30421

amyeroberts merged 6 commits into huggingface:main from EduardoPach:fix-seggpt-loss

Fixed main train issues

5b001b80

EduardoPach force pushed to 5b001b80 1 year ago

amyeroberts commented on 2024-04-23

Added loss test

34be5d66

Update src/transformers/models/seggpt/modeling_seggpt.py

c2746764

Added missing labels arg in SegGptModel forward

b89941aa

Fixed typo

db8addaa

Added slow test to test loss calculation

8a74151b

amyeroberts approved these changes on 2024-04-24

amyeroberts merged d26c1413 into main 1 year ago

Reviewers

amyeroberts

Assignees

No one assigned

Labels

None yet

Milestone

No milestone