package option

type AiOption struct {
	AdapterId             string
	ClusterIds            []string
	TaskName              string
	TaskId                int64
	Replica               int32
	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
	ModelName             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{}

	Output string

	ResourcesRequired []map[string]interface{}
}

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