Skip to content

Commit

Permalink
Merge pull request pytorch#569 from jspisak/patch-4
Browse files Browse the repository at this point in the history
Update PyTorch Cheat.md
  • Loading branch information
brianjo committed Jul 22, 2019
2 parents a4a2fbb + 81520da commit 9f21916
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions beginner_source/PyTorch Cheat.md
Original file line number Diff line number Diff line change
Expand Up @@ -26,13 +26,13 @@ from torch.jit import script, trace # hybrid frontend decorator and tracin
```
See [autograd](https://pytorch.org/docs/stable/autograd.html), [nn](https://pytorch.org/docs/stable/nn.html), [functional](https://pytorch.org/docs/stable/nn.html#torch-nn-functional) and [optim](https://pytorch.org/docs/stable/optim.html)

### Hybrid frontend
### Torchscript and JIT

```
torch.jit.trace() # takes your module or function and an example data input, and traces the computational steps that the data encounters as it progresses through the model
@script # decorator used to indicate data-dependent control flow within the code being traced
```
See [hybrid frontend](https://pytorch.org/docs/stable/hybridfrontend)
See [Torchscript](https://pytorch.org/docs/stable/jit.html)

### ONNX

Expand Down

0 comments on commit 9f21916

Please sign in to comment.