[fix] Fix UNITER unit tests on gpu

facebookresearch / mmf

A modular framework for vision & language multimodal research from Facebook AI Research (FAIR)

https://mmf.sh/

Other

5.49k stars 935 forks source link

Closed Ryan-Qiyu-Jiang closed 2 years ago

Ryan-Qiyu-Jiang commented 2 years ago

Stack from ghstack:

Send sample list to cuda device when available.

Differential Revision: D33002411

Ryan-Qiyu-Jiang commented 2 years ago

@ryan-qiyu-jiang has imported this pull request. If you are a Facebook employee, you can view this diff on Phabricator.

facebook-github-bot commented 2 years ago

@Ryan-Qiyu-Jiang has updated the pull request. You must reimport the pull request before landing.

Ryan-Qiyu-Jiang commented 2 years ago

@ryan-qiyu-jiang has imported this pull request. If you are a Facebook employee, you can view this diff on Phabricator.