전이학습 상세 로직 수정

This commit is contained in:
2026-02-11 14:08:21 +09:00
parent 224ddae68b
commit 92de48b55e

View File

@@ -81,6 +81,7 @@ public class ModelTrainDetailService {
datasetIds.add(mappingDataset.getDatasetId()); datasetIds.add(mappingDataset.getDatasetId());
} }
datasetReq.setIds(datasetIds); datasetReq.setIds(datasetIds);
datasetReq.setModelNo(modelInfo.getModelNo());
if (modelInfo.getModelNo().equals("G1")) { if (modelInfo.getModelNo().equals("G1")) {
dataSets = mngCoreService.getDatasetSelectG1List(datasetReq); dataSets = mngCoreService.getDatasetSelectG1List(datasetReq);