Browse Source

mod

opt-modelarts
lewis 3 years ago
parent
commit
8d638bb4a9
1 changed files with 2 additions and 0 deletions
  1. +2
    -0
      modules/modelarts/modelarts.go

+ 2
- 0
modules/modelarts/modelarts.go View File

@@ -521,6 +521,7 @@ func GenerateTrainJobVersion(ctx *context.Context, req *GenerateTrainJobReq, job

createTime := timeutil.TimeStampNow()
task := &models.Cloudbrain{
//todo: mod status and job_id
Status: models.JobStatusTemp,
UserID: ctx.User.ID,
RepoID: ctx.Repo.Repository.ID,
@@ -853,6 +854,7 @@ func InitSpecialPool() {
}
}

//1、修改方案:当创建任务502的时候写入temp表,后续轮询temp表,如果查询到记录,则先停止,后删除。
func HandleTrainJobInfo(task *models.Cloudbrain) error {
if isTempJob(task.JobID, task.Status) {
if task.VersionCount > VersionCountOne {


Loading…
Cancel
Save