riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-08-03 03:31:05 +06:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity
d26c14139c
transformers/tests/models/seggpt
History
Eduardo Pacheco d26c14139c
[SegGPT] Fix loss calculation (#30421) 
* Fixed main train issues

* Added loss test

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

Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com>

* Added missing labels arg in SegGptModel forward

* Fixed typo

* Added slow test to test loss calculation

---------

Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com>
2024-04-24 15:24:34 +01:00
..
__init__.py Adding SegGPT (#27735) 2024-02-26 18:17:19 +00:00
test_image_processing_seggpt.py Adding SegGPT (#27735) 2024-02-26 18:17:19 +00:00
test_modeling_seggpt.py [SegGPT] Fix loss calculation (#30421) 2024-04-24 15:24:34 +01:00