pytorch

Documentation

Index

• func ParseElasticConfig(elasticConfig ElasticConfig) *kubeflowv1.ElasticPolicy
• type ElasticConfig

Functions

func ParseElasticConfig

func ParseElasticConfig(elasticConfig ElasticConfig) *kubeflowv1.ElasticPolicy

To support parsing elastic config from both v0 and v1 of kubeflow pytorch idl

Types

type ElasticConfig

type ElasticConfig interface {
	GetMinReplicas() int32
	GetMaxReplicas() int32
	GetNprocPerNode() int32
	GetMaxRestarts() int32
	GetRdzvBackend() string
}

Interface for unified elastic config handling across plugin version v0 and v1. This interface should always be aligned with the ElasticConfig defined in flyteidl.