transformers/tests/models/bridgetower
Anahita Bhiwandiwalla 16121bae5c
Update BridgeTowerForContrastiveLearning (#22145)
* Use return_loss for BridgeTowerForContrastiveLearning, add example

* fix tests

* Update example in BridgeTowerForContrastiveLearning

* Update test_modeling_bridgetower.py

* update model output format

* minor update

* Update src/transformers/models/bridgetower/modeling_bridgetower.py

* make style

---------

Co-authored-by: Tiep Le <97980157+tileintel@users.noreply.github.com>
Co-authored-by: Tiep Le <tiep.le@intel.com>
Co-authored-by: Yih-Dar <2521628+ydshieh@users.noreply.github.com>
Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
2023-03-15 20:54:38 +01:00
..
__init__.py Add BridgeTower model (#20775) 2023-01-25 14:04:32 -05:00
test_image_processing_bridgetower.py Add BridgeTower model (#20775) 2023-01-25 14:04:32 -05:00
test_modeling_bridgetower.py Update BridgeTowerForContrastiveLearning (#22145) 2023-03-15 20:54:38 +01:00