Add support for more generic inputs to "Recursive copy to"

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

Add support for more generic inputs to "Recursive copy to" #653

Closed agordo closed 3 years ago

agordo commented 3 years ago

Summary: Add support to other types to recursive_copy_to_device. In particular, as long as the input has a "to" method, recursive_copy_to_device will be able to handle it. This opens support to more complex inputs, e.g. classes.

Differential Revision: D24910581

facebook-github-bot commented 3 years ago

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

facebook-github-bot commented 3 years ago

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