2 Commits

View File

@@ -22,6 +22,8 @@ public class TrainJobWorker {
private final ModelTrainJobCoreService modelTrainJobCoreService;
private final ModelTrainMngCoreService modelTrainMngCoreService;
private final ModelTrainMetricsJobService modelTrainMetricsJobService;
private final ModelTestMetricsJobService modelTestMetricsJobService;
private final DockerTrainService dockerTrainService;
private final ObjectMapper objectMapper;
@@ -90,8 +92,10 @@ public class TrainJobWorker {
if (isEval) {
modelTrainMngCoreService.markStep2Success(modelId);
modelTrainMetricsJobService.findTrainValidMetricCsvFiles();
} else {
modelTrainMngCoreService.markStep1Success(modelId);
modelTestMetricsJobService.findTestValidMetricCsvFiles();
}
} else {