frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

[functorch] updated README

Browse Source
This commit is contained in:
Horace He
2021-05-07 14:14:02 -07:00
committed by Jon Janzen
parent ac017463e0
commit 26d2ab7e35
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
functorch/README.md
View File

@ -78,7 +78,7 @@ Run a quick sanity check in python:
`functorch` is a PyTorch C++ Extension module. To install,
- Install [PyTorch from source](https://github.com/pytorch/pytorch#from-source).
8134806e23 is a known commit that this project works with; commits after that
3d2ce60 is a known commit that this project works with; commits after that
should be fine but haven't been formally tested.
- Run `python setup.py install`. You can use `DEBUG=1` to compile in debug mode.
@ -97,7 +97,7 @@ Right now, we support the following transforms:
Furthermore, we have some utilities for working with PyTorch modules.
- `make_functional(model)` takes a model and returns its weights and a function
version of the model that has no state.
- `make_functional_with_buffers(model)` takes a model and returns its weights
- `make_functional_with_buffers(model)` takes a model and returns its weights
and buffers and a function version of the model that has no state.
### vmap