add ckpt

routers/repo/cloudbrain.go

@@ -177,6 +177,12 @@ func cloudBrainVersionNewDataPrepare(ctx *context.Context) error {
 	waitCount := cloudbrain.GetWaitingCloudbrainCount(models.TypeCloudBrainOne, models.GPUResource, models.JobTypeTrain)
 	ctx.Data["wait_count"] = waitCount
 
+	ctx.Data["model_name"] = ctx.Cloudbrain.ModelName
+	ctx.Data["label_name"] = ctx.Cloudbrain.LabelName
+	ctx.Data["ckpt_name"] = ctx.Cloudbrain.CkptName
+	ctx.Data["model_version"] = ctx.Cloudbrain.ModelVersion
+	ctx.Data["pre_train_model_url"] = ctx.Cloudbrain.PreTrainModelUrl
+
 	return nil
 }
 

routers/repo/grampus.go

@@ -258,6 +258,12 @@ func grampusTrainJobVersionNewDataPrepare(ctx *context.Context, processType stri
 		ctx.Data["work_server_number"] = ctx.Cloudbrain.WorkServerNumber
 	}
 
+	ctx.Data["model_name"] = ctx.Cloudbrain.ModelName
+	ctx.Data["label_name"] = ctx.Cloudbrain.LabelName
+	ctx.Data["ckpt_name"] = ctx.Cloudbrain.CkptName
+	ctx.Data["model_version"] = ctx.Cloudbrain.ModelVersion
+	ctx.Data["pre_train_model_url"] = ctx.Cloudbrain.PreTrainModelUrl
+
 	return nil
 }