tests/models/seggpt/test_modeling_seggpt.py · d26c14139c373b2265a4759775687b67e9fb03eb · chenpangpang / transformers

[SegGPT] Fix loss calculation (#30421) · d26c1413

Eduardo Pacheco authored Apr 24, 2024



* 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>

d26c1413

test_modeling_seggpt.py 17.6 KB

Replace test_modeling_seggpt.py