JoYang
/
pcm-coordinator
forked from JointCloud/pcm-coordinator
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge branch 'master' of https://gitlink.org.cn/JointCloud/pcm-coordinator

This commit is contained in:
zhangwei 2024-11-11 17:29:12 +08:00
parent c5e6d67d06 61a2513a51
commit 8c08e49cea
1 changed files with 17 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
internal/storeLink/storeLink.go
View File

@ -73,6 +73,11 @@ const (
)
var (
Datasets = []string{"mnist", "cifar10"}
AlgorithmsForDatasets = map[string][]string{
"mnist": {"fcn"},
"cifar10": {"cnn"},
}
OctImgStatus = map[int32]string{
1: "未上传",
3: "制作完成",
@ -153,6 +158,10 @@ func GetModelNamesByType(t string) ([]string, error) {
}
func GetDatasetsNames(ctx context.Context, collectorMap map[string]collector.AiCollector) ([]string, error) {
return Datasets, nil
}
func GetDatasetsNamesSync(ctx context.Context, collectorMap map[string]collector.AiCollector) ([]string, error) {
var wg sync.WaitGroup
var errCh = make(chan interface{}, len(collectorMap))
var errs []interface{}
@ -225,6 +234,14 @@ func GetDatasetsNames(ctx context.Context, collectorMap map[string]collector.AiC
}
func GetAlgorithms(ctx context.Context, collectorMap map[string]collector.AiCollector, resourceType string, taskType string, dataset string) ([]string, error) {
algorithm := AlgorithmsForDatasets[dataset]
if len(algorithm) != 0 {
return algorithm, nil
}
return nil, errors.New("not found")
}
func GetAlgorithmsSync(ctx context.Context, collectorMap map[string]collector.AiCollector, resourceType string, taskType string, dataset string) ([]string, error) {
var names []string
var wg sync.WaitGroup
var errCh = make(chan interface{}, len(collectorMap))