Customize model forward

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

Customize model forward #764

Closed edpizzi closed 3 years ago

edpizzi commented 3 years ago

Summary: Minor refactor to introduce a compute_model method, similar to the compute_loss method. This allows customizing the model forward pass in subclasses.

Differential Revision: D30647461

facebook-github-bot commented 3 years ago

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

facebook-github-bot commented 3 years ago

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