forked from JointCloud/pcm-coordinator
45 lines
993 B
Go
45 lines
993 B
Go
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
|
|
}
|