transformers/tests/models/bridgetower
Anahita Bhiwandiwalla 4cb5ffa93d
Add loss for BridgeTowerForMaskedLM and BridgeTowerForImageAndTextRetrieval (#21684)
* Add loss for BridgeTowerForMaskedLM and BridgeTowerForImageAndTextRetrieval

* minor fix return_dict

* implement test for loss computation

---------

Co-authored-by: Tiep Le <97980157+tileintel@users.noreply.github.com>
Co-authored-by: Tiep Le <tiep.le@intel.com>
2023-02-28 12:21:48 -05: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 Add loss for BridgeTowerForMaskedLM and BridgeTowerForImageAndTextRetrieval (#21684) 2023-02-28 12:21:48 -05:00