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