transformers
4a66c0d9 - enable training mask2former and maskformer for transformers trainer (#28277)

Commit

2 years ago

enable training mask2former and maskformer for transformers trainer (#28277) * fix get_num_masks output as [int] to int * fix loss size from torch.Size([1]) to torch.Size([])

References

#28277 - enable training mask2former and maskformer for transformers trainer

Author

SangbumChoi

Parents

6b8ec258

transformers 4a66c0d9 - enable training mask2former and maskformer for transformers trainer (#28277)

transformers
4a66c0d9 - enable training mask2former and maskformer for transformers trainer (#28277)