2.0.0b53

flyteplugins.pytorch.task

Directory

Class	Description
`Elastic`	Elastic defines the configuration for running a PyTorch elastic job using torch.
`RunPolicy`	RunPolicy describes some policy to apply to the execution of a kubeflow job.
`TorchFunctionTask`	Plugin to transform local python code for execution as a PyTorch job.

Method	Description
`launcher_entrypoint()`

def launcher_entrypoint(
    tctx: flyte.models.TaskContext,
    fn: bytes,
    kwargs: **kwargs,
)