transformers/tests/models/mask2former
Sangbum Daniel Choi 4a66c0d952
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([])
2024-01-04 09:53:25 +01:00
..
__init__.py
test_image_processing_mask2former.py [Styling] stylify using ruff (#27144) 2023-11-16 17:43:19 +01:00
test_modeling_mask2former.py enable training mask2former and maskformer for transformers trainer (#28277) 2024-01-04 09:53:25 +01:00