https://github.com/facebookresearch/pythia
Revision 273a56b903d4bcd73467ddaa3605cf147311471d authored by Vedanuj Goswami on 30 June 2021, 21:14:48 UTC, committed by Facebook GitHub Bot on 30 June 2021, 21:15:33 UTC
Summary:
PR adds Image-text matching loss to MMF. The loss can be used standalone or with the MMFTransformer as a head.

Addresses https://github.com/facebookresearch/mmf/issues/947 #466

- Adds ITM loss
- Fixes ITM loss for validation dataset
- Adds config for COCO ITM loss
- Adds tests for ITM loss

Pull Request resolved: https://github.com/facebookresearch/mmf/pull/961

Test Plan:
Imported from GitHub, without a `Test Plan:` line.

Pass tests

Reviewed By: apsdehal

Differential Revision: D28925014

Pulled By: vedanuj

fbshipit-source-id: d123468f746820e1412380384dd15c5ebed15887
1 parent a13040b
History
Tip revision: 273a56b903d4bcd73467ddaa3605cf147311471d authored by Vedanuj Goswami on 30 June 2021, 21:14:48 UTC
[feat] ITM Loss (#961)
Tip revision: 273a56b
File Mode Size
.circleci
.github
docs
mmf
mmf_cli
projects
tests
tools
website
.editorconfig -rw-r--r-- 191 bytes
.flake8 -rw-r--r-- 187 bytes
.gitignore -rw-r--r-- 267 bytes
.pre-commit-config.yaml -rw-r--r-- 1.1 KB
LICENSE -rw-r--r-- 1.5 KB
MANIFEST.in -rw-r--r-- 130 bytes
NOTICES -rw-r--r-- 4.5 KB
README.md -rw-r--r-- 2.2 KB
pyproject.toml -rw-r--r-- 1017 bytes
requirements.txt -rw-r--r-- 483 bytes
setup.py -rw-r--r-- 5.1 KB

README.md

back to top