https://github.com/facebookresearch/pythia
Revision 6395711ac07c9aa1a238cc595b1aaa7f8a3d94a2 authored by Amanpreet Singh on 31 March 2021, 20:47:25 UTC, committed by Facebook GitHub Bot on 31 March 2021, 20:48:30 UTC
Summary:
Pull Request resolved: https://github.com/facebookresearch/mmf/pull/842

Recent changes for datamodules support moved the options num_workers and pin_memory along with others inside the dataset_config. This diff fixes backwards compatibility and allows specifying the num_workers and pin_memory from training config.

Reviewed By: ytsheng, vedanuj, rayhou0710

Differential Revision: D27421166

fbshipit-source-id: 1fff516dcbf6a0f8bb5ab81ca6c477b53b0e4937
1 parent 03a0527
Raw File
Tip revision: 6395711ac07c9aa1a238cc595b1aaa7f8a3d94a2 authored by Amanpreet Singh on 31 March 2021, 20:47:25 UTC
[fix] Reallow num_workers and pin_memory overrides from training config (#842)
Tip revision: 6395711
requirements.txt
torch==1.6.0
torchvision==0.7.0
numpy>=1.16.6
tqdm>=4.43.0,<4.50.0
demjson==2.2.4
torchtext==0.5.0
GitPython==3.1.0
requests==2.23.0
fasttext==0.9.1
nltk==3.4.5
editdistance==0.5.3
transformers==3.4.0
sklearn==0.0
omegaconf==2.0.6
lmdb==0.98
termcolor==1.1.0
iopath==0.1.7
pytorch_lightning==1.1.6
datasets==1.2.1
matplotlib==3.3.4
pycocotools==2.0.2
back to top