|
|
|
@@ -6,10 +6,8 @@ import ( |
|
|
|
"gitlink.org.cn/jcce-pcm/pcm-coordinator/api/internal/svc" |
|
|
|
"gitlink.org.cn/jcce-pcm/pcm-coordinator/pkg/models" |
|
|
|
"gitlink.org.cn/jcce-pcm/pcm-coordinator/pkg/utils" |
|
|
|
"gitlink.org.cn/jcce-pcm/pcm-coordinator/pkg/utils/timeutils" |
|
|
|
"gitlink.org.cn/jcce-pcm/pcm-participant-ac/hpcAC" |
|
|
|
"strings" |
|
|
|
"time" |
|
|
|
) |
|
|
|
|
|
|
|
type ShuguangAi struct { |
|
|
|
@@ -74,8 +72,6 @@ func (s *ShuguangAi) SubmitTask(imageId string, cmd string, params []string, res |
|
|
|
return nil, err |
|
|
|
} |
|
|
|
|
|
|
|
dateStr := timeutils.UnixTimeToString(time.Now().Unix()) |
|
|
|
|
|
|
|
//python参数 |
|
|
|
var pythonArg string |
|
|
|
for _, param := range params { |
|
|
|
@@ -85,7 +81,7 @@ func (s *ShuguangAi) SubmitTask(imageId string, cmd string, params []string, res |
|
|
|
|
|
|
|
req := &hpcAC.SubmitPytorchTaskReq{ |
|
|
|
Params: &hpcAC.SubmitPytorchTaskParams{ |
|
|
|
TaskName: TASK_PYTORCH_PREFIX + UNDERSCORE + utils.RandomString(7) + UNDERSCORE + dateStr, |
|
|
|
TaskName: TASK_PYTORCH_PREFIX + UNDERSCORE + utils.RandomString(17), |
|
|
|
WorkPath: WorkPath, |
|
|
|
IsDistributed: false, |
|
|
|
IsHvd: false, |
|
|
|
|