Skip to content

Commit d99b39c

Browse files
committed
fixed the quick start example for pytorch
1 parent 4c48ddf commit d99b39c

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

docs/source/user_guide/model_registration/quick_start.rst

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -139,8 +139,8 @@ Create a model, prepare it, verify that it works, save it to the model catalog,
139139
artifact_dir=tempfile.mkdtemp()
140140
torch_model = PyTorchModel(torch_estimator, artifact_dir=artifact_dir)
141141
142-
# Autogenerate score.py, pickled model, runtime.yaml, input_schema.json and output_schema.json
143-
# Set `save_entire_model` to `True` to save the model as Torchscript program.
142+
# Autogenerate score.py, serialized model, runtime.yaml
143+
# Set `use_torch_script` to `True` to save the model as Torchscript program.
144144
torch_model.prepare(inference_conda_env="pytorch110_p38_cpu_v1", use_torch_script=True)
145145
146146
# Verify generated artifacts

0 commit comments

Comments
 (0)