Fix broken torchscript test

facebookresearch / ClassyVision

An end-to-end PyTorch framework for image and video classification

https://classyvision.ai

MIT License

1.59k stars 277 forks source link

Closed mannatsingh closed 4 years ago

mannatsingh commented 4 years ago

Summary: The test fails on GPUs since the model isn't moved over to the GPU, fixed that

Differential Revision: D24699001

facebook-github-bot commented 4 years ago

This pull request was exported from Phabricator. Differential Revision: D24699001

facebook-github-bot commented 4 years ago

This pull request was exported from Phabricator. Differential Revision: D24699001