Skip to content

Pytorch

Initialization of the PyTorch integration.

PytorchIntegration

Bases: Integration

Definition of PyTorch integration for ZenML.

Source code in src/zenml/integrations/pytorch/__init__.py 
22
23
24
25
26
27
28
29
30
31
class PytorchIntegration(Integration):
    """Definition of PyTorch integration for ZenML."""

    NAME = PYTORCH
    REQUIREMENTS = ["torch"]

    @classmethod
    def activate(cls) -> None:
        """Activates the integration."""
        from zenml.integrations.pytorch import materializers  # noqa

activate() classmethod

Activates the integration.

Source code in src/zenml/integrations/pytorch/__init__.py 
28
29
30
31
@classmethod
def activate(cls) -> None:
    """Activates the integration."""
    from zenml.integrations.pytorch import materializers  # noqa