PyTorchModel

PyTorch model creation step

SageMaker step type	CreateModel
Node type	internal (consumes upstream, produces downstream)
Interface file	steps/interfaces/pytorch_model.step.yaml

Compute

Compute kind	model
SDK class	PyTorchModel (SageMaker DLC via image_uris.retrieve)

Functionality

SageMaker Model creation step for PyTorch. No script — managed by SageMaker ModelStep. Creates a deployable model from training artifacts.

Inputs (dependencies)

Input	Type	Required	Compatible producers
model_data	model_artifacts	yes	PyTorchTraining, ProcessingStep, ModelArtifactsStep, Package

Outputs

Output	Type
model_name	custom_property

Consumers (downstream steps)

Steps that declare this step as a compatible input source:

• BatchTransform

• LightGBMMTModelInference

• LightGBMModelEval

• LightGBMModelInference

• PyTorchModelEval

• PyTorchModelInference

• SlipboxKnowledgeRouting

• TSAModelEval

• XGBoostModelEval

• XGBoostModelInference

---

← Back to the Step Catalog