pcm-coordinator/api/internal/scheduler/schedulers/option/aiOption.go

package option

type AiOption struct {
	AiClusterId           string // shuguangAi /octopus ClusterId
	TaskName              string
	ResourceType          string // cpu/gpu/compute card
	CpuCoreNum            int64
	TaskType              string // pytorch/tensorflow/mindspore
	DatasetsName          string // mnist/imageNet/iris
	StrategyName          string
	ClusterToStaticWeight map[string]int32
	Tops                  float64
	ComputeCard           string
	CodeType              string
	AlgorithmName         string

	ImageId     string
	SpecId      string
	DatasetsId  string
	CodeId      string
	ResourceId  string
	AlgorithmId string

	Cmd    string
	Envs   []string
	Params []string

	Datasets      string
	AlgorithmCode string
	Image         string
	Model         interface{}
}

func (a AiOption) GetOptionType() string {
	return AI
}