torch

package

v2.1.0-rc.1 Latest Latest Go to latest Published: Nov 3, 2025 License: Apache-2.0 Imports: 18 Imported by: 0

Details

View Source

const Name = "Torch"

This section is empty.

func New(context.Context, client.Client, client.FieldIndexer) (framework.Plugin, error)

type Torch struct{}

func (t *Torch) EnforceMLPolicy(info *runtime.Info, trainJob *trainer.TrainJob) error

TODO (andreyvelich): Add support for PyTorch elastic when JobSet supports Elastic Jobs.

func (t *Torch) Name() string

func (t *Torch) Validate(_ context.Context, runtimeInfo *runtime.Info, _, newObj *trainer.TrainJob) (admission.Warnings, field.ErrorList)