PyTorchSpec defines arguments for configuring PyTorch model serving.
Name | Type | Description | Notes |
---|---|---|---|
model_class_name | str | Defaults PyTorch model class name to 'PyTorchModel' | [optional] |
resources | V1ResourceRequirements | Defaults to requests and limits of 1CPU, 2Gb MEM. | [optional] |
runtime_version | str | PyTorch KFServer docker image version which defaults to latest release | [optional] |
storage_uri | str | The URI of the trained model which contains model.pt |