flytekitplugins.kfpytorch.PyTorch#

class flytekitplugins.kfpytorch.PyTorch(num_workers)[source]#

Configuration for an executable Pytorch Job. Use this to run distributed pytorch training on k8s

Parameters
  • num_workers (int) – integer determining the number of worker replicas spawned in the cluster for this job

  • master). ((in addition to 1) –

Return type

None

Methods

Attributes

num_workers: int#