transformers/tests/models/segformer
NielsRogge 2875fa971c
[SegFormer] Add support for segmentation masks with one label (#20279)
* Add support for binary segmentation

* Fix loss calculation and add test

* Remove space

* use fstring

Co-authored-by: Niels Rogge <nielsrogge@Nielss-MacBook-Pro.local>
Co-authored-by: Niels Rogge <nielsrogge@Nielss-MBP.localdomain>
2022-12-20 16:46:50 +01:00
..
__init__.py Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
test_feature_extraction_segformer.py Add Image Processors (#19796) 2022-11-02 11:57:36 +00:00
test_modeling_segformer.py [SegFormer] Add support for segmentation masks with one label (#20279) 2022-12-20 16:46:50 +01:00
test_modeling_tf_segformer.py Run tests if skip condition not met (#18764) 2022-08-30 14:03:28 +02:00