search

pytorch / functorch

functorch is JAX-like composable function transforms for PyTorch.
https://pytorch.org/functorch/
BSD 3-Clause "New" or "Revised" License
1.38k stars 102 forks source link

Add test_make_fx_model_train example #980

Closed mostafaelhoushi closed 1 year ago

mostafaelhoushi commented 1 year ago

Adding examples that import fx traces of full training graph (forward pass, backward pass, and weight updates)

facebook-github-bot commented 1 year ago

@mostafaelhoushi has imported this pull request. If you are a Meta employee, you can view this diff on Phabricator.

facebook-github-bot commented 1 year ago

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

facebook-github-bot commented 1 year ago

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

facebook-github-bot commented 1 year ago

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

facebook-github-bot commented 1 year ago

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

facebook-github-bot commented 1 year ago

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

facebook-github-bot commented 1 year ago

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

facebook-github-bot commented 1 year ago

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