Compare commits
2 Commits
452494d44d
...
83204abfe9
| Author | SHA1 | Date | |
|---|---|---|---|
| 83204abfe9 | |||
| 5b682c1386 |
@@ -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 {
|
||||
|
||||
Reference in New Issue
Block a user