sagemaker.pytorch.rst

PyTorch

PyTorch Estimator

.. autoclass:: sagemaker.pytorch.estimator.PyTorch
    :members:
    :undoc-members:
    :show-inheritance:

PyTorch Model

.. autoclass:: sagemaker.pytorch.model.PyTorchModel
    :members:
    :undoc-members:
    :show-inheritance:

PyTorch Predictor

.. autoclass:: sagemaker.pytorch.model.PyTorchPredictor
    :members:
    :undoc-members:
    :show-inheritance: