0.0.0+develop
Classes
| Class | Description |
|---|---|
flytekitplugins.kfpytorch.task.Elastic |
Configuration for `torch elastic training <https://pytorch. |
flytekitplugins.kfpytorch.task.ElasticWorkerResult |
A named tuple representing the result of a torch elastic worker process. |
flytekitplugins.kfpytorch.task.Master |
Configuration for master replica group. |
flytekitplugins.kfpytorch.task.PyTorch |
Configuration for an executable `PyTorch Job <https://github. |
flytekitplugins.kfpytorch.task.PyTorchFunctionTask |
Plugin that submits a PyTorchJob (see https://github. |
flytekitplugins.kfpytorch.task.PytorchElasticFunctionTask |
Plugin for distributed training with torch elastic/torchrun (see. |
flytekitplugins.kfpytorch.task.RunPolicy |
RunPolicy describes some policy to apply to the execution of a kubeflow job. |
flytekitplugins.kfpytorch.task.Worker |