| @@ -59,7 +59,7 @@ public class ExperimentInstanceStatusTask { | |||||
| if (trainMetricRecords != null && !trainMetricRecords.isEmpty()) { | if (trainMetricRecords != null && !trainMetricRecords.isEmpty()) { | ||||
| for (Map<String, Object> trainMetricRecord : trainMetricRecords) { | for (Map<String, Object> trainMetricRecord : trainMetricRecords) { | ||||
| String taskId = (String) trainMetricRecord.get("task_id"); | String taskId = (String) trainMetricRecord.get("task_id"); | ||||
| if (taskId.startsWith("model-export")) { | |||||
| if (taskId.startsWith("model-train")) { | |||||
| String runId = (String) trainMetricRecord.get("run_id"); | String runId = (String) trainMetricRecord.get("run_id"); | ||||
| List<InsMetricInfoVo> expTrainInfos = aimService.getExpTrainInfos1(true, experimentIns.getExperimentId(), runId); | List<InsMetricInfoVo> expTrainInfos = aimService.getExpTrainInfos1(true, experimentIns.getExperimentId(), runId); | ||||
| for (InsMetricInfoVo expTrainInfo : expTrainInfos) { | for (InsMetricInfoVo expTrainInfo : expTrainInfos) { | ||||
| @@ -73,7 +73,7 @@ public class ExperimentInstanceStatusTask { | |||||
| if (evaluateMetricRecords != null && !evaluateMetricRecords.isEmpty()) { | if (evaluateMetricRecords != null && !evaluateMetricRecords.isEmpty()) { | ||||
| for (Map<String, Object> evaluateMetricRecord : evaluateMetricRecords) { | for (Map<String, Object> evaluateMetricRecord : evaluateMetricRecords) { | ||||
| String taskId = (String) evaluateMetricRecord.get("task_id"); | String taskId = (String) evaluateMetricRecord.get("task_id"); | ||||
| if (taskId.startsWith("model-export")) { | |||||
| if (taskId.startsWith("model-evaluate")) { | |||||
| String runId = (String) evaluateMetricRecord.get("run_id"); | String runId = (String) evaluateMetricRecord.get("run_id"); | ||||
| List<InsMetricInfoVo> expTrainInfos = aimService.getExpTrainInfos1(false, experimentIns.getExperimentId(), runId); | List<InsMetricInfoVo> expTrainInfos = aimService.getExpTrainInfos1(false, experimentIns.getExperimentId(), runId); | ||||
| for (InsMetricInfoVo expTrainInfo : expTrainInfos) { | for (InsMetricInfoVo expTrainInfo : expTrainInfos) { | ||||