use the cuda to represent the gpu device

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 lly-zero-one closed 4 years ago

lly-zero-one commented 4 years ago

Summary: self.device is set as "cuda" when the device is gpu.

Differential Revision: D24645064

facebook-github-bot commented 4 years ago

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

facebook-github-bot commented 4 years ago

This pull request has been merged in facebookresearch/ClassyVision@897d9be5e98b1e4359ab418b044f1235553edae3.