transformers
enable training mask2former and maskformer for transformers trainer
#28277

Merged

enable training mask2former and maskformer for transformers trainer #28277

ArthurZucker merged 2 commits into huggingface:main from SangbumChoi:fix_mask2former

fix get_num_masks output as [int] to int

902c5be0

fix loss size from torch.Size([1]) to torch.Size([])

c8a8a492

NielsRogge approved these changes on 2024-01-03

NielsRogge requested a review from

ArthurZucker 2 years ago

ArthurZucker approved these changes on 2024-01-04

ArthurZucker merged 4a66c0d9 into main 2 years ago

SangbumChoi deleted the fix_mask2former branch 1 year ago

Reviewers

ArthurZucker

NielsRogge

Assignees

No one assigned

Labels

None yet

Milestone

No milestone