training

Documentation

Index

• Constants
• Variables
• func ConfigureRoutes(routeGroup *gin.RouterGroup, config config.ModelTrainingConfig, ...)
• func ConfigureToolchainRoutes(routeGroup *gin.RouterGroup, tiService toolchainService)
• type ModelTrainingController
• type MtValidator
  • func NewMtValidator(tiService toolchainGetter, connRepository conn_repository.Repository, ...) *MtValidator
  • func (mtv *MtValidator) ValidatesAndSetDefaults(mt *training.ModelTraining) (err error)
• type TiValidator
  • func NewTiValidator() *TiValidator
  • func (tiv *TiValidator) ValidatesAndSetDefaults(ti *training.ToolchainIntegration) (err error)
• type ToolchainIntegrationController

Constants

const (
	GetModelTrainingURL        = "/model/training/:id"
	GetAllModelTrainingURL     = "/model/training"
	GetModelTrainingLogsURL    = "/model/training/:id/log"
	CreateModelTrainingURL     = "/model/training"
	UpdateModelTrainingURL     = "/model/training"
	SaveModelTrainingResultURL = "/model/training/:id/result"
	DeleteModelTrainingURL     = "/model/training/:id"
	IDMtURLParam               = "id"
	FollowURLParam             = "follow"
)

const (
	MtVcsNotExistsErrorMessage           = "cannot find VCS Connection"
	MtObjectStorageNotExistsErrorMessage = "cannot find ObjectStorage Connection"
	EmptyModelNameErrorMessage           = "empty model.name"
	EmptyModelVersionErrorMessage        = "empty model.version"
	ValidationMtErrorMessage             = "Validation of model training is failed"
	WrongVcsTypeErrorMessage             = "VCS connection must have the GIT type. You pass the connection of %s type"
	EmptyAlgorithmSourceNameMessageError = "both VCS and ObjectStorage names are empty"
	MultipleAlgorithmSourceMessageError  = "both VCS and ObjectStorage names are specified, must be only one"
	EmptyDataBindingNameErrorMessage     = "you should specify connection name for %d number of data binding"
	EmptyDataBindingPathErrorMessage     = "you should specify local path for %d number of data binding"
	WrongDataBindingTypeErrorMessage     = "%s data binding has wrong data type. Currently supported the following types" +
		" of connections for data bindings: %v"
	ToolchainEmptyErrorMessage         = "empty toolchain parameter"
	WrongObjectStorageTypeErrorMessage = "%s object storage has wrong data type. Currently supported the following types" +
		" of connections for object storage: %v"
	UnknownNodeSelector = "node selector %v is not presented in ODAHU config"
)

const (
	GetToolchainIntegrationURL    = "/toolchain/integration/:id"
	GetAllToolchainIntegrationURL = "/toolchain/integration"
	CreateToolchainIntegrationURL = "/toolchain/integration"
	UpdateToolchainIntegrationURL = "/toolchain/integration"
	DeleteToolchainIntegrationURL = "/toolchain/integration/:id"
	IDTiURLParam                  = "id"
)

const (
	ValidationTiErrorMessage      = "Validation of toolchain integration is failed"
	EmptyEntrypointErrorMessage   = "empty entrypoint"
	EmptyDefaultImageErrorMessage = "empty defaultImage"
)

Variables

var (
	DefaultArtifactOutputTemplate = "{{ .Name }}-{{ .Version }}-{{ .RandomUUID }}.zip"
	DefaultWorkDir                = "./"
)

Functions

func ConfigureRoutes

func ConfigureRoutes(
	routeGroup *gin.RouterGroup,
	config config.ModelTrainingConfig,
	gpuResourceName string,
	trainService mt_service.Service,
	toolchainService toolchainGetter,
	connRepo conn_repository.Repository,
	trainKubeClient mt_kube_client.Client)

func ConfigureToolchainRoutes

func ConfigureToolchainRoutes(
	routeGroup *gin.RouterGroup,
	tiService toolchainService,
)

Types

type ModelTrainingController

type ModelTrainingController struct {
	// contains filtered or unexported fields
}

type MtValidator

type MtValidator struct {
	// contains filtered or unexported fields
}

func NewMtValidator

func NewMtValidator(
	tiService toolchainGetter,
	connRepository conn_repository.Repository,
	trainingConfig config.ModelTrainingConfig,
	gpuResourceName string,
) *MtValidator

func (*MtValidator) ValidatesAndSetDefaults

func (mtv *MtValidator) ValidatesAndSetDefaults(mt *training.ModelTraining) (err error)

type TiValidator

type TiValidator struct {
}

func NewTiValidator

func NewTiValidator() *TiValidator

func (*TiValidator) ValidatesAndSetDefaults

func (tiv *TiValidator) ValidatesAndSetDefaults(ti *training.ToolchainIntegration) (err error)

type ToolchainIntegrationController

type ToolchainIntegrationController struct {
	// contains filtered or unexported fields
}