From 4752033910ed48ad3c6a9fe3cccce61574468f92 Mon Sep 17 00:00:00 2001
From: ychao_1983 <ychao_1983@sina.com>
Date: Wed, 14 Sep 2022 15:07:53 +0800
Subject: [PATCH] =?UTF-8?q?=E6=8F=90=E4=BA=A4=E4=BB=A3=E7=A0=81?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 routers/repo/grampus.go | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/routers/repo/grampus.go b/routers/repo/grampus.go
index ed869e76e..dee8fbdd1 100755
--- a/routers/repo/grampus.go
+++ b/routers/repo/grampus.go
@@ -887,6 +887,10 @@ func generateCommand(repoName, processorType, codeRemotePath, dataRemotePath, bo
 		}
 	}
 
+	if pretrainModelFileName != "" {
+		paramCode += " --pretrainmodelname" + "=" + pretrainModelFileName
+	}
+
 	var commandCode string
 	if processorType == grampus.ProcessorTypeNPU {
 		commandCode = "/bin/bash /home/work/run_train_for_openi.sh " + workDir + "code/" + strings.ToLower(repoName) + "/" + bootFile + " /tmp/log/train.log" + paramCode + ";"