Skip to content

Commit feee20f

Browse files
committed
remove readme changes
1 parent 4fc3c63 commit feee20f

File tree

2 files changed

+18
-0
lines changed

2 files changed

+18
-0
lines changed

README.md

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -323,6 +323,16 @@ trainer = Trainer(callbacks=[checkpointing])
323323

324324
</details>
325325

326+
<details>
327+
<summary>Export to torchscript (JIT) (production use)</summary>
328+
329+
```python
330+
# torchscript
331+
autoencoder = LitAutoEncoder()
332+
torch.jit.save(autoencoder.to_torchscript(), "model.pt")
333+
```
334+
335+
</details>
326336

327337
<details>
328338
<summary>Export to ONNX (production use)</summary>

src/pytorch_lightning/README.md

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -271,6 +271,14 @@ checkpointing = ModelCheckpoint(monitor="val_loss")
271271
trainer = Trainer(callbacks=[checkpointing])
272272
```
273273

274+
<summary>Export to torchscript (JIT) (production use)</summary>
275+
276+
```python
277+
# torchscript
278+
autoencoder = LitAutoEncoder()
279+
torch.jit.save(autoencoder.to_torchscript(), "model.pt")
280+
```
281+
274282
<summary>Export to ONNX (production use)</summary>
275283

276284
```python

0 commit comments

Comments
 (0)