File tree Expand file tree Collapse file tree 2 files changed +18
-0
lines changed
Expand file tree Collapse file tree 2 files changed +18
-0
lines changed Original file line number Diff line number Diff 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 >
Original file line number Diff line number Diff line change @@ -271,6 +271,14 @@ checkpointing = ModelCheckpoint(monitor="val_loss")
271271trainer = 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
You can’t perform that action at this time.
0 commit comments