pytorch

package
v1.9.16 Latest Latest
Warning

This package is not in the latest version of its module.

Go to latest
Published: Oct 11, 2023 License: Apache-2.0 Imports: 19 Imported by: 1

Documentation

Index

Constants

This section is empty.

Variables

This section is empty.

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.

Jump to

Keyboard shortcuts

? : This menu
/ : Search site
f or F : Jump to
y or Y : Canonical URL