https://github.com/facebookresearch/pythia
Tip revision: 63c18a1640c91f861717a63b6a2b60384ee3882b authored by Amanpreet Singh on 29 September 2021, 18:58:24 UTC
[feat] FSDP support for MMF
[feat] FSDP support for MMF
Tip revision: 63c18a1
File | Mode | Size |
---|---|---|
albef | ||
interfaces | ||
transformers | ||
unit | ||
__init__.py | -rw-r--r-- | 1.1 KB |
alignment.py | -rw-r--r-- | 9.2 KB |
ban.py | -rw-r--r-- | 3.2 KB |
base_model.py | -rw-r--r-- | 14.5 KB |
butd.py | -rw-r--r-- | 7.1 KB |
cnn_lstm.py | -rw-r--r-- | 3.5 KB |
frcnn.py | -rw-r--r-- | 9.2 KB |
fusions.py | -rw-r--r-- | 7.2 KB |
krisp.py | -rw-r--r-- | 30.7 KB |
lorra.py | -rw-r--r-- | 2.2 KB |
lxmert.py | -rw-r--r-- | 27.7 KB |
m4c.py | -rw-r--r-- | 21.8 KB |
m4c_captioner.py | -rw-r--r-- | 748 bytes |
mmbt.py | -rw-r--r-- | 25.1 KB |
mmf_bert.py | -rw-r--r-- | 16.0 KB |
mmf_transformer.py | -rw-r--r-- | 17.2 KB |
movie_mcan.py | -rw-r--r-- | 9.8 KB |
pythia.py | -rw-r--r-- | 18.6 KB |
top_down_bottom_up.py | -rw-r--r-- | 2.5 KB |
unimodal.py | -rw-r--r-- | 3.5 KB |
vilbert.py | -rw-r--r-- | 57.1 KB |
visdial_multi_modal.py | -rw-r--r-- | 3.3 KB |
visual_bert.py | -rw-r--r-- | 23.7 KB |