diff --git a/src/main/java/com/kamco/cd/training/model/service/ModelTrainMngService.java b/src/main/java/com/kamco/cd/training/model/service/ModelTrainMngService.java index 9567a5c..a5a4496 100644 --- a/src/main/java/com/kamco/cd/training/model/service/ModelTrainMngService.java +++ b/src/main/java/com/kamco/cd/training/model/service/ModelTrainMngService.java @@ -12,7 +12,6 @@ import com.kamco.cd.training.model.dto.ModelTrainMngDto; import com.kamco.cd.training.model.dto.ModelTrainMngDto.SearchReq; import com.kamco.cd.training.postgres.core.HyperParamCoreService; import com.kamco.cd.training.postgres.core.ModelTrainMngCoreService; -import java.io.IOException; import java.util.List; import java.util.UUID; import lombok.RequiredArgsConstructor; @@ -93,22 +92,6 @@ public class ModelTrainMngService { // 모델 config 저장 modelTrainMngCoreService.saveModelConfig(modelId, req.getModelConfig()); - UUID tmpUuid = UUID.randomUUID(); - String raw = tmpUuid.toString().toUpperCase().replace("-", ""); - - List uids = - modelTrainMngCoreService.findDatasetUid(req.getTrainingDataset().getDatasetList()); - - try { - // 데이터셋 심볼링크 생성 - String tmpUid = tmpDatasetService.buildTmpDatasetSymlink(raw, uids); - ModelTrainMngDto.UpdateReq updateReq = new ModelTrainMngDto.UpdateReq(); - updateReq.setRequestPath(tmpUid); - modelTrainMngCoreService.updateModelMaster(modelId, updateReq); - } catch (IOException e) { - throw new RuntimeException(e); - } - return modelUuid; } diff --git a/src/main/java/com/kamco/cd/training/postgres/core/ModelTrainMngCoreService.java b/src/main/java/com/kamco/cd/training/postgres/core/ModelTrainMngCoreService.java index c6d3c91..35dcb68 100644 --- a/src/main/java/com/kamco/cd/training/postgres/core/ModelTrainMngCoreService.java +++ b/src/main/java/com/kamco/cd/training/postgres/core/ModelTrainMngCoreService.java @@ -111,16 +111,8 @@ public class ModelTrainMngCoreService { entity.setTrainType(addReq.getTrainType()); // 일반, 전이 entity.setBeforeModelId(addReq.getBeforeModelId()); - if (addReq.getIsStart()) { - entity.setModelStep((short) 1); - entity.setStatusCd(TrainStatusType.IN_PROGRESS.getId()); - entity.setStrtDttm(ZonedDateTime.now()); - entity.setStep1StrtDttm(ZonedDateTime.now()); - entity.setStep1State(TrainStatusType.IN_PROGRESS.getId()); - } else { - entity.setStatusCd(TrainStatusType.READY.getId()); - entity.setStep1State(TrainStatusType.READY.getId()); - } + entity.setStatusCd(TrainStatusType.READY.getId()); + entity.setStep1State(TrainStatusType.READY.getId()); entity.setCreatedUid(userUtil.getId()); ModelMasterEntity resultEntity = modelMngRepository.save(entity); diff --git a/src/main/java/com/kamco/cd/training/train/service/TrainJobService.java b/src/main/java/com/kamco/cd/training/train/service/TrainJobService.java index 66a1851..df9f876 100644 --- a/src/main/java/com/kamco/cd/training/train/service/TrainJobService.java +++ b/src/main/java/com/kamco/cd/training/train/service/TrainJobService.java @@ -40,6 +40,7 @@ public class TrainJobService { private String responseDir; public Long getModelIdByUuid(UUID uuid) { + createTmpFile(uuid); return modelTrainMngCoreService.findModelIdByUuid(uuid); }