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