torch

Documentation

Index

• Constants
• func New(context.Context, client.Client, client.FieldIndexer) (framework.Plugin, error)
• type Torch
  • func (t *Torch) EnforceMLPolicy(info *runtime.Info, trainJob *kubeflowv2.TrainJob) error
  • func (t *Torch) Name() string
  • func (t *Torch) Validate(oldObj, newObj *kubeflowv2.TrainJob) (admission.Warnings, field.ErrorList)

Constants

const Name = "Torch"

Functions

func New

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

Types

type Torch

type Torch struct{}

func (*Torch) EnforceMLPolicy

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

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

func (*Torch) Name

func (t *Torch) Name() string

func (*Torch) Validate

func (t *Torch) Validate(oldObj, newObj *kubeflowv2.TrainJob) (admission.Warnings, field.ErrorList)

TODO: Need to implement validateions for TorchJob.