facebookresearch / ClassyVision

An end-to-end PyTorch framework for image and video classification
https://classyvision.ai
MIT License
1.59k stars 278 forks source link

ClassyModelWrapper as nn.Module #805

Open msaroufim opened 1 year ago

msaroufim commented 1 year ago

Summary: Fixes https://fb.workplace.com/groups/1148364849411890/permalink/1278947843020256/

Differential Revision: D45665663

facebook-github-bot commented 1 year ago

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