param opt

modules/cloudbrain/cloudbrain.go

@@ -21,6 +21,7 @@ const (
 	CodeMountPath         = "/code"
 	DataSetMountPath      = "/dataset"
 	ModelMountPath        = "/model"
+	LogFile				  = "log.txt"
 	BenchMarkMountPath    = "/benchmark"
 	BenchMarkResourceID   = 1
 	Snn4imagenetMountPath = "/snn4imagenet"

routers/repo/cloudbrain.go

@@ -1439,6 +1439,7 @@ func getTrainJobCommand(form auth.CreateCloudBrainForm) (string, error) {
 	}
 
 	var parameters models.Parameters
+	var param string
 	if len(params) != 0 {
 		err := json.Unmarshal([]byte(params), &parameters)
 		if err != nil {
@@ -1447,11 +1448,11 @@ func getTrainJobCommand(form auth.CreateCloudBrainForm) (string, error) {
 		}
 
 		for _, parameter := range parameters.Parameter {
-			command += "export " + parameter.Label + "=" + parameter.Value + ";"
+			param += " --" + parameter.Label + "=" + parameter.Value
 		}
 	}
 
-	command += "python /code/" + bootFile
+	command += "python /code/" + bootFile + param + " > " + cloudbrain.ModelMountPath + "/" + cloudbrain.LogFile
 
 	return command, nil
 }