Commit 030e9760 authored by linpeiqin's avatar linpeiqin

小bug

parent def4bcbe
...@@ -125,7 +125,7 @@ public class ModelEstimateController { ...@@ -125,7 +125,7 @@ public class ModelEstimateController {
array.add(jsonObject.get("maximumGeneratingLength")); array.add(jsonObject.get("maximumGeneratingLength"));
array.add(jsonObject.get("ToppSamplingValue")); array.add(jsonObject.get("ToppSamplingValue"));
array.add(jsonObject.get("temperatureCoefficient")); array.add(jsonObject.get("temperatureCoefficient"));
array.add(this.pythonConfig.getModelEstimateFileBaseDir() + File.separator + modelVersion.getModelUrl() + File.separator + "evl_" + taskId); array.add(this.pythonConfig.getModelEstimateFileBaseDir() + modelVersion.getVersionName() + File.separator + "evl_" + taskId);
System.out.println(array.toJSONString()); System.out.println(array.toJSONString());
return ResponseResult.success(array.toJSONString()); return ResponseResult.success(array.toJSONString());
} }
......
...@@ -198,6 +198,7 @@ public class TuningRunServiceImpl extends BaseService<TuningRun, Long> implement ...@@ -198,6 +198,7 @@ public class TuningRunServiceImpl extends BaseService<TuningRun, Long> implement
modelManageS.setModelDescribe(runPublishDto.getModelDescribe()); modelManageS.setModelDescribe(runPublishDto.getModelDescribe());
modelManageS.setModelName(runPublishDto.getModelName()); modelManageS.setModelName(runPublishDto.getModelName());
modelManageS.setModelType(runPublishDto.getModelType()); modelManageS.setModelType(runPublishDto.getModelType());
modelManageS.setIsBaseModel(0);
modelVersionURl = pythonConfig.getModelOutputFileBaseDir() + runPublishDto.getModelName() + "_V1"; modelVersionURl = pythonConfig.getModelOutputFileBaseDir() + runPublishDto.getModelName() + "_V1";
modelVersionS.setModelUrl(modelVersionURl); modelVersionS.setModelUrl(modelVersionURl);
this.modelManageService.saveAndCreateVersion(modelManageS, modelVersionS); this.modelManageService.saveAndCreateVersion(modelManageS, modelVersionS);
...@@ -234,7 +235,7 @@ public class TuningRunServiceImpl extends BaseService<TuningRun, Long> implement ...@@ -234,7 +235,7 @@ public class TuningRunServiceImpl extends BaseService<TuningRun, Long> implement
if (receiveMsg.equals("send_data")) { if (receiveMsg.equals("send_data")) {
JSONArray array = new JSONArray(); JSONArray array = new JSONArray();
array.add("zh"); array.add("zh");
array.add(modelManage.getModelName()); array.add(modelVersion.getVersionName());
array.add(modelVersion.getModelUrl()); array.add(modelVersion.getModelUrl());
JSONArray ddJson = new JSONArray(); JSONArray ddJson = new JSONArray();
ddJson.add("train_" + tuningRun.getRunId()); ddJson.add("train_" + tuningRun.getRunId());
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment