hyperparam_with_modeltype

Reviewed-on: #55

build.gradle

@@ -3,6 +3,7 @@ plugins {
   id 'org.springframework.boot' version '3.5.7'
   id 'io.spring.dependency-management' version '1.1.7'
   id 'com.diffplug.spotless' version '6.25.0'
+  id 'idea'
 }
 
 group = 'com.kamco.cd'
@@ -21,11 +22,23 @@ configurations {
   }
 }
 
+// QueryDSL 생성된 소스 디렉토리 정의
+def generatedSourcesDir = file("$buildDir/generated/sources/annotationProcessor/java/main")
+
 repositories {
   mavenCentral()
   maven { url "https://repo.osgeo.org/repository/release/" }
 }
 
+// Gradle이 생성된 소스를 컴파일 경로에 포함하도록 설정
+sourceSets {
+  main {
+    java {
+      srcDirs += generatedSourcesDir
+    }
+  }
+}
+
 dependencies {
   implementation 'org.springframework.boot:spring-boot-starter-data-jpa'
   implementation 'org.springframework.boot:spring-boot-starter-web'
@@ -87,6 +100,21 @@ dependencies {
   implementation 'org.apache.commons:commons-csv:1.10.0'
 }
 
+// IntelliJ가 생성된 소스를 인식하도록 설정
+idea {
+  module {
+    // 소스 디렉토리로 인식
+    sourceDirs += generatedSourcesDir
+
+    // Generated Sources Root로 마킹 (IntelliJ에서 특별 처리)
+    generatedSourceDirs += generatedSourcesDir
+
+    // 소스 및 Javadoc 다운로드
+    downloadJavadoc = true
+    downloadSources = true
+  }
+}
+
 configurations.configureEach {
   exclude group: 'javax.media', module: 'jai_core'
 }
@@ -95,6 +123,21 @@ tasks.named('test') {
   useJUnitPlatform()
 }
 
+// 컴파일 전 생성된 소스 디렉토리 생성 보장
+tasks.named('compileJava') {
+  doFirst {
+    generatedSourcesDir.mkdirs()
+  }
+}
+
+// 생성된 소스 정리 태스크
+tasks.register('cleanGeneratedSources', Delete) {
+  delete generatedSourcesDir
+}
+
+tasks.named('clean') {
+  dependsOn 'cleanGeneratedSources'
+}
 
 bootJar {
   archiveFileName = 'ROOT.jar'

docker-compose-dev.yml

@@ -14,6 +14,7 @@ services:
       - /mnt/nfs_share/images:/app/original-images
       - /mnt/nfs_share/model_output:/app/model-outputs
       - /mnt/nfs_share/train_dataset:/app/train-dataset
+      - /home/kcomu/data:/home/kcomu/data
     networks:
       - kamco-cds
     restart: unless-stopped

src/main/java/com/kamco/cd/training/common/dto/HyperParam.java

@@ -1,5 +1,6 @@
 package com.kamco.cd.training.common.dto;
 
+import com.kamco.cd.training.common.enums.ModelType;
 import io.swagger.v3.oas.annotations.media.Schema;
 import lombok.AllArgsConstructor;
 import lombok.Getter;
@@ -14,6 +15,10 @@ public class HyperParam {
   // -------------------------
   // Important
   // -------------------------
+
+  @Schema(description = "모델", example = "large")
+  private ModelType model; // backbone
+
   @Schema(description = "백본 네트워크", example = "large")
   private String backbone; // backbone
 

src/main/java/com/kamco/cd/training/common/enums/ModelType.java

@@ -2,6 +2,7 @@ package com.kamco.cd.training.common.enums;
 
 import com.kamco.cd.training.common.utils.enums.CodeExpose;
 import com.kamco.cd.training.common.utils.enums.EnumType;
+import java.util.Arrays;
 import lombok.AllArgsConstructor;
 import lombok.Getter;
 
@@ -15,6 +16,10 @@ public enum ModelType implements EnumType {
 
   private String desc;
 
+  public static ModelType getValueData(String modelNo) {
+    return Arrays.stream(ModelType.values()).filter(m -> m.getId().equals(modelNo)).findFirst().orElse(G1);
+  }
+
   @Override
   public String getId() {
     return name();

src/main/java/com/kamco/cd/training/common/utils/FIleChecker.java

@@ -506,11 +506,15 @@ public class FIleChecker {
 
     try {
       File dir = new File(targetPath);
+      log.info("targetPath={}", targetPath);
+      log.info("absolute targetPath={}", dir.getAbsolutePath());
+
       if (!dir.exists()) {
         dir.mkdirs();
       }
 
       File dest = new File(dir, String.valueOf(chunkIndex));
+      log.info("real save path = {}", dest.getAbsolutePath());
 
       log.info("chunkIndex={}, uploadSize={}", chunkIndex, mfile.getSize());
       log.info("savedSize={}", dest.length());
@@ -522,6 +526,9 @@ public class FIleChecker {
       log.info("after delete={}", dest.length());
       mfile.transferTo(dest);
 
+      log.info("after transfer size={}", dest.length());
+      log.info("after transfer exists={}", dest.exists());
+
       return true;
     } catch (IOException e) {
       log.error("chunk save error", e);

src/main/java/com/kamco/cd/training/dataset/DatasetApiController.java

@@ -14,15 +14,11 @@ import io.swagger.v3.oas.annotations.responses.ApiResponse;
 import io.swagger.v3.oas.annotations.responses.ApiResponses;
 import io.swagger.v3.oas.annotations.tags.Tag;
 import jakarta.validation.Valid;
-import java.nio.file.Path;
-import java.nio.file.Paths;
 import java.util.List;
 import java.util.UUID;
 import lombok.RequiredArgsConstructor;
 import org.springframework.core.io.Resource;
-import org.springframework.core.io.UrlResource;
 import org.springframework.data.domain.Page;
-import org.springframework.http.MediaType;
 import org.springframework.http.ResponseEntity;
 import org.springframework.web.bind.annotation.*;
 
@@ -230,11 +226,7 @@ public class DatasetApiController {
       throws Exception {
 
     String path = datasetService.getFilePathByUUIDPathType(uuid, pathType);
-    Path filePath = Paths.get(path);
-
-    Resource resource = new UrlResource(filePath.toUri());
-
-    return ResponseEntity.ok().contentType(MediaType.APPLICATION_OCTET_STREAM).body(resource);
+    return datasetService.getFilePathByFile(path);
   }
 
   @Operation(summary = "객체별 파일 Path 조회", description = "파일 Path 조회")

src/main/java/com/kamco/cd/training/dataset/service/DatasetService.java

@@ -4,9 +4,6 @@ import com.fasterxml.jackson.databind.JsonNode;
 import com.fasterxml.jackson.databind.ObjectMapper;
 import com.fasterxml.jackson.databind.node.ArrayNode;
 import com.fasterxml.jackson.databind.node.ObjectNode;
-import com.jcraft.jsch.ChannelSftp;
-import com.jcraft.jsch.JSch;
-import com.jcraft.jsch.Session;
 import com.kamco.cd.training.common.enums.LearnDataType;
 import com.kamco.cd.training.common.exception.CustomApiException;
 import com.kamco.cd.training.common.service.FormatStorage;
@@ -32,7 +29,6 @@ import java.util.ArrayList;
 import java.util.HashMap;
 import java.util.List;
 import java.util.Map;
-import java.util.Properties;
 import java.util.UUID;
 import java.util.stream.Stream;
 import lombok.RequiredArgsConstructor;
@@ -502,29 +498,9 @@ public class DatasetService {
 
   public ResponseEntity<Resource> getFilePathByFile(String remoteFilePath) {
 
-    String host = "192.168.2.86";
-    String user = "kcomu";
-    String password = "Kamco2025!";
-
-    Session session = null;
-    ChannelSftp sftp = null;
-
     try {
-      JSch jsch = new JSch();
-
-      session = jsch.getSession(user, host, 22);
-      session.setPassword(password);
-
-      Properties config = new Properties();
-      config.put("StrictHostKeyChecking", "no");
-      session.setConfig(config);
-      session.connect(10_000);
-
-      sftp = (ChannelSftp) session.openChannel("sftp");
-      sftp.connect(5_000);
-
-      // 86 서버 파일을 스트림으로 연다
-      InputStream inputStream = sftp.get(normalizeLinuxPath(remoteFilePath));
+      Path path = Paths.get(remoteFilePath);
+      InputStream inputStream = Files.newInputStream(path);
 
       InputStreamResource resource =
           new InputStreamResource(inputStream) {

src/main/java/com/kamco/cd/training/hyperparam/HyperParamApiController.java

@@ -1,6 +1,7 @@
 package com.kamco.cd.training.hyperparam;
 
 import com.kamco.cd.training.common.dto.HyperParam;
+import com.kamco.cd.training.common.enums.ModelType;
 import com.kamco.cd.training.config.api.ApiResponseDto;
 import com.kamco.cd.training.hyperparam.dto.HyperParamDto;
 import com.kamco.cd.training.hyperparam.dto.HyperParamDto.List;
@@ -65,7 +66,7 @@ public class HyperParamApiController {
                     mediaType = "application/json",
                     schema = @Schema(implementation = String.class))),
         @ApiResponse(responseCode = "400", description = "잘못된 요청", content = @Content),
-        @ApiResponse(responseCode = "422", description = "HPs_0001 수정 불가", content = @Content),
+        @ApiResponse(responseCode = "422", description = "default는 삭제불가", content = @Content),
         @ApiResponse(responseCode = "500", description = "서버 오류", content = @Content)
       })
   @PutMapping("/{uuid}")
@@ -87,8 +88,9 @@ public class HyperParamApiController {
         @ApiResponse(responseCode = "404", description = "하이퍼파라미터를 찾을 수 없음", content = @Content),
         @ApiResponse(responseCode = "500", description = "서버 오류", content = @Content)
       })
-  @GetMapping("/list")
+  @GetMapping("{model}/list")
   public ApiResponseDto<Page<List>> getHyperParam(
+      @PathVariable ModelType model,
       @Parameter(
               description = "구분 CREATE_DATE(생성일), LAST_USED_DATE(최근사용일)",
               example = "CREATE_DATE")
@@ -98,7 +100,7 @@ public class HyperParamApiController {
           LocalDate startDate,
       @Parameter(description = "종료일", example = "2026-02-28") @RequestParam(required = false)
           LocalDate endDate,
-      @Parameter(description = "버전명", example = "HPs_0001") @RequestParam(required = false)
+      @Parameter(description = "버전명", example = "G_000001") @RequestParam(required = false)
           String hyperVer,
       @Parameter(
               description = "정렬",
@@ -124,7 +126,7 @@ public class HyperParamApiController {
     searchReq.setSort(sort);
     searchReq.setPage(page);
     searchReq.setSize(size);
-    Page<List> list = hyperParamService.getHyperParamList(searchReq);
+    Page<List> list = hyperParamService.getHyperParamList(model, searchReq);
 
     return ApiResponseDto.ok(list);
   }
@@ -133,7 +135,7 @@ public class HyperParamApiController {
   @ApiResponses(
       value = {
         @ApiResponse(responseCode = "200", description = "삭제 성공", content = @Content),
-        @ApiResponse(responseCode = "422", description = "HPs_0001 삭제 불가", content = @Content),
+        @ApiResponse(responseCode = "422", description = "default 삭제 불가", content = @Content),
         @ApiResponse(responseCode = "404", description = "하이퍼파라미터를 찾을 수 없음", content = @Content),
       })
   @DeleteMapping("/{uuid}")
@@ -179,8 +181,11 @@ public class HyperParamApiController {
         @ApiResponse(responseCode = "404", description = "하이퍼파라미터를 찾을 수 없음", content = @Content),
         @ApiResponse(responseCode = "500", description = "서버 오류", content = @Content)
       })
-  @GetMapping("/init")
-  public ApiResponseDto<HyperParamDto.Basic> getInitHyperParam() {
-    return ApiResponseDto.ok(hyperParamService.getInitHyperParam());
+  @GetMapping("/init/{model}")
+  public ApiResponseDto<HyperParamDto.Basic> getInitHyperParam(
+    @PathVariable ModelType model
+
+  ) {
+    return ApiResponseDto.ok(hyperParamService.getInitHyperParam(model));
   }
 }

src/main/java/com/kamco/cd/training/hyperparam/dto/HyperParamDto.java

@@ -1,5 +1,6 @@
 package com.kamco.cd.training.hyperparam.dto;
 
+import com.kamco.cd.training.common.enums.ModelType;
 import com.kamco.cd.training.common.utils.enums.CodeExpose;
 import com.kamco.cd.training.common.utils.enums.EnumType;
 import com.kamco.cd.training.common.utils.interfaces.JsonFormatDttm;
@@ -24,6 +25,7 @@ public class HyperParamDto {
   @AllArgsConstructor
   public static class Basic {
 
+    private ModelType model; // 20250212 modeltype추가
     private UUID uuid;
     private String hyperVer;
     @JsonFormatDttm private ZonedDateTime createdDttm;
@@ -98,6 +100,8 @@ public class HyperParamDto {
     private Integer gpuCnt;
     private String gpuIds;
     private Integer masterPort;
+
+    private Boolean isDefault;
   }
 
   @Getter

src/main/java/com/kamco/cd/training/hyperparam/service/HyperParamService.java

@@ -1,8 +1,10 @@
 package com.kamco.cd.training.hyperparam.service;
 
 import com.kamco.cd.training.common.dto.HyperParam;
+import com.kamco.cd.training.common.enums.ModelType;
 import com.kamco.cd.training.hyperparam.dto.HyperParamDto;
 import com.kamco.cd.training.hyperparam.dto.HyperParamDto.List;
+import com.kamco.cd.training.hyperparam.dto.HyperParamDto.SearchReq;
 import com.kamco.cd.training.postgres.core.HyperParamCoreService;
 import java.util.UUID;
 import lombok.RequiredArgsConstructor;
@@ -20,11 +22,12 @@ public class HyperParamService {
   /**
    * 하이퍼 파라미터 목록 조회
    *
+   * @param model
    * @param req
    * @return 목록
    */
-  public Page<List> getHyperParamList(HyperParamDto.SearchReq req) {
-    return hyperParamCoreService.findByHyperVerList(req);
+  public Page<List> getHyperParamList(ModelType model, SearchReq req) {
+    return hyperParamCoreService.findByHyperVerList(model, req);
   }
 
   /**
@@ -59,8 +62,8 @@ public class HyperParamService {
   }
 
   /** 하이퍼파라미터 최적화 설정값 조회 */
-  public HyperParamDto.Basic getInitHyperParam() {
-    return hyperParamCoreService.getInitHyperParam();
+  public HyperParamDto.Basic getInitHyperParam(ModelType model) {
+    return hyperParamCoreService.getInitHyperParam(model);
   }
 
   /**

src/main/java/com/kamco/cd/training/model/ModelTrainMngApiController.java

@@ -74,13 +74,12 @@ public class ModelTrainMngApiController {
   @ApiResponses(
       value = {
         @ApiResponse(responseCode = "200", description = "삭제 성공", content = @Content),
-        @ApiResponse(responseCode = "409", description = "HPs_0001 삭제 불가", content = @Content)
+        @ApiResponse(responseCode = "409", description = "G1_000001 삭제 불가", content = @Content)
       })
   @DeleteMapping("/{uuid}")
   public ApiResponseDto<Void> deleteModelTrain(
       @Parameter(description = "학습 모델 uuid", example = "f2b02229-90f2-45f5-92ea-c56cf1c29f79")
-          @PathVariable
-          UUID uuid) {
+          @PathVariable UUID uuid) {
     modelTrainMngService.deleteModelTrain(uuid);
     return ApiResponseDto.ok(null);
   }
@@ -92,9 +91,8 @@ public class ModelTrainMngApiController {
         @ApiResponse(responseCode = "500", description = "서버 오류", content = @Content)
       })
   @PostMapping
-  public ApiResponseDto<String> createModelTrain(@Valid @RequestBody ModelTrainMngDto.AddReq req) {
-    modelTrainMngService.createModelTrain(req);
-    return ApiResponseDto.ok("ok");
+  public ApiResponseDto<UUID> createModelTrain(@Valid @RequestBody ModelTrainMngDto.AddReq req) {
+    return ApiResponseDto.ok(modelTrainMngService.createModelTrain(req));
   }
 
   @Operation(summary = "모델학습 config 정보 조회", description = "모델학습 config 정보 조회 API")
@@ -150,4 +148,22 @@ public class ModelTrainMngApiController {
     req.setDataType(selectType);
     return ApiResponseDto.ok(modelTrainMngService.getDatasetSelectList(req));
   }
+
+  @Operation(summary = "모델학습 1단계 실행중인 것이 있는지 count", description = "모델학습 1단계 실행중인 것이 있는지 count")
+  @ApiResponses(
+      value = {
+        @ApiResponse(
+            responseCode = "200",
+            description = "검색 성공",
+            content =
+                @Content(
+                    mediaType = "application/json",
+                    schema = @Schema(implementation = Long.class))),
+        @ApiResponse(responseCode = "400", description = "잘못된 검색 조건", content = @Content),
+        @ApiResponse(responseCode = "500", description = "서버 오류", content = @Content)
+      })
+  @GetMapping("/ing-training-cnt")
+  public ApiResponseDto<Long> findModelStep1InProgressCnt() {
+    return ApiResponseDto.ok(modelTrainMngService.findModelStep1InProgressCnt());
+  }
 }

src/main/java/com/kamco/cd/training/model/dto/ModelTrainMngDto.java

@@ -155,6 +155,17 @@ public class ModelTrainMngDto {
     ModelConfig modelConfig;
   }
 
+  @Schema(name = "addReq", description = "모델학습 관리 등록 파라미터")
+  @Getter
+  @Setter
+  @NoArgsConstructor
+  @AllArgsConstructor
+  public static class UpdateReq {
+
+    private String requestPath;
+    private String responsePath;
+  }
+
   @Getter
   @Setter
   public static class TrainingDataset {

src/main/java/com/kamco/cd/training/model/service/ModelTrainMngService.java

@@ -12,7 +12,8 @@ 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 com.kamco.cd.training.train.service.TrainJobService;
+import java.io.IOException;
+import java.nio.file.Path;
 import java.util.List;
 import java.util.UUID;
 import lombok.RequiredArgsConstructor;
@@ -28,9 +29,9 @@ import org.springframework.transaction.annotation.Transactional;
 @Slf4j
 public class ModelTrainMngService {
 
-  private final TrainJobService trainJobService;
   private final ModelTrainMngCoreService modelTrainMngCoreService;
   private final HyperParamCoreService hyperParamCoreService;
+  private final TmpDatasetService tmpDatasetService;
 
   /**
    * 모델학습 조회
@@ -59,7 +60,7 @@ public class ModelTrainMngService {
    * @return
    */
   @Transactional
-  public void createModelTrain(ModelTrainMngDto.AddReq req) {
+  public UUID createModelTrain(ModelTrainMngDto.AddReq req) {
     HyperParam hyperParam = req.getHyperParam();
     HyperParamDto.Basic hyper = new HyperParamDto.Basic();
 
@@ -78,7 +79,10 @@ public class ModelTrainMngService {
     }
 
     // 모델학습 테이블 저장
-    Long modelId = modelTrainMngCoreService.saveModel(req);
+    ModelTrainMngDto.Basic modelDto = modelTrainMngCoreService.saveModel(req);
+
+    Long modelId = modelDto.getId();
+    UUID modelUuid = modelDto.getUuid();
 
     // 모델학습 데이터셋 저장
     modelTrainMngCoreService.saveModelDataset(modelId, req);
@@ -90,10 +94,23 @@ public class ModelTrainMngService {
     // 모델 config 저장
     modelTrainMngCoreService.saveModelConfig(modelId, req.getModelConfig());
 
-    // 저장 다 끝난 뒤에 job enqueue
-    if (Boolean.TRUE.equals(req.getIsStart())) {
-      trainJobService.enqueue(modelId); // job 저장 + 이벤트 발행(실행은 AFTER_COMMIT)
+    UUID tmpUuid = UUID.randomUUID();
+    String raw = tmpUuid.toString().toUpperCase().replace("-", "");
+
+    List<String> uids =
+        modelTrainMngCoreService.findDatasetUid(req.getTrainingDataset().getDatasetList());
+
+    try {
+      // 데이터셋 심볼링크 생성
+      Path path = tmpDatasetService.buildTmpDatasetSymlink(raw, uids);
+      ModelTrainMngDto.UpdateReq updateReq = new ModelTrainMngDto.UpdateReq();
+      updateReq.setRequestPath(path.toString());
+      modelTrainMngCoreService.updateModelMaster(modelId, updateReq);
+    } catch (IOException e) {
+      throw new RuntimeException(e);
     }
+
+    return modelUuid;
   }
 
   /**
@@ -119,4 +136,8 @@ public class ModelTrainMngService {
       return modelTrainMngCoreService.getDatasetSelectG2G3List(req);
     }
   }
+
+  public Long findModelStep1InProgressCnt() {
+    return modelTrainMngCoreService.findModelStep1InProgressCnt();
+  }
 }

src/main/java/com/kamco/cd/training/model/service/TmpDatasetService.java

@@ -0,0 +1,66 @@
+package com.kamco.cd.training.model.service;
+
+import java.io.IOException;
+import java.nio.file.*;
+import java.util.List;
+import lombok.RequiredArgsConstructor;
+import lombok.extern.slf4j.Slf4j;
+import org.springframework.beans.factory.annotation.Value;
+import org.springframework.stereotype.Service;
+import org.springframework.transaction.annotation.Transactional;
+
+@Slf4j
+@Service
+@RequiredArgsConstructor
+public class TmpDatasetService {
+
+  @Value("${train.docker.requestDir}")
+  private String requestDir;
+
+  @Transactional(readOnly = true)
+  public Path buildTmpDatasetSymlink(String uid, List<String> datasetUids) throws IOException {
+
+    // 환경에 맞게 yml로 빼는 걸 추천
+    Path BASE = Paths.get(requestDir);
+    Path tmp = BASE.resolve("tmp").resolve(uid);
+
+    // mkdir -p "$TMP"/train/{input1,input2,label} ...
+    for (String type : List.of("train", "val")) {
+      for (String part : List.of("input1", "input2", "label")) {
+        Files.createDirectories(tmp.resolve(type).resolve(part));
+      }
+    }
+
+    for (String id : datasetUids) {
+      Path srcRoot = BASE.resolve(id);
+
+      for (String type : List.of("train", "val")) {
+        for (String part : List.of("input1", "input2", "label")) {
+
+          Path srcDir = srcRoot.resolve(type).resolve(part);
+
+          // zsh NULL_GLOB: 폴더가 없으면 그냥 continue
+          if (!Files.isDirectory(srcDir)) continue;
+
+          try (DirectoryStream<Path> stream = Files.newDirectoryStream(srcDir)) {
+            for (Path f : stream) {
+              if (!Files.isRegularFile(f)) continue;
+
+              String dstName = id + "__" + f.getFileName();
+              Path dst = tmp.resolve(type).resolve(part).resolve(dstName);
+
+              // 이미 있으면 스킵(원하면 덮어쓰기 로직으로 바꿀 수 있음)
+              if (Files.exists(dst)) continue;
+
+              // ln -s "$f" "$dst" 와 동일
+              Files.createSymbolicLink(dst, f.toAbsolutePath());
+            }
+          }
+        }
+      }
+    }
+
+    log.info("tmp dataset created: {}", tmp);
+    return tmp;
+  }
+}

src/main/java/com/kamco/cd/training/postgres/core/HyperParamCoreService.java

@@ -1,10 +1,12 @@
 package com.kamco.cd.training.postgres.core;
 
 import com.kamco.cd.training.common.dto.HyperParam;
+import com.kamco.cd.training.common.enums.ModelType;
 import com.kamco.cd.training.common.exception.CustomApiException;
 import com.kamco.cd.training.common.utils.UserUtil;
 import com.kamco.cd.training.hyperparam.dto.HyperParamDto;
 import com.kamco.cd.training.hyperparam.dto.HyperParamDto.Basic;
+import com.kamco.cd.training.hyperparam.dto.HyperParamDto.SearchReq;
 import com.kamco.cd.training.postgres.entity.ModelHyperParamEntity;
 import com.kamco.cd.training.postgres.repository.hyperparam.HyperParamRepository;
 import java.time.ZonedDateTime;
@@ -17,6 +19,7 @@ import org.springframework.stereotype.Service;
 @Service
 @RequiredArgsConstructor
 public class HyperParamCoreService {
+
   private final HyperParamRepository hyperParamRepository;
   private final UserUtil userUtil;
 
@@ -27,7 +30,7 @@ public class HyperParamCoreService {
    * @return 등록된 버전명
    */
   public Basic createHyperParam(HyperParam createReq) {
-    String firstVersion = getFirstHyperParamVersion();
+    String firstVersion = getFirstHyperParamVersion(createReq.getModel());
 
     ModelHyperParamEntity entity = new ModelHyperParamEntity();
     entity.setHyperVer(firstVersion);
@@ -47,17 +50,17 @@ public class HyperParamCoreService {
   /**
    * 하이퍼파라미터 수정
    *
-   * @param uuid uuid
+   * @param uuid      uuid
    * @param createReq 등록 요청
    * @return ver
    */
   public String updateHyperParam(UUID uuid, HyperParam createReq) {
     ModelHyperParamEntity entity =
-        hyperParamRepository
-            .findHyperParamByUuid(uuid)
-            .orElseThrow(() -> new CustomApiException("NOT_FOUND_DATA", HttpStatus.NOT_FOUND));
+      hyperParamRepository
+        .findHyperParamByUuid(uuid)
+        .orElseThrow(() -> new CustomApiException("NOT_FOUND_DATA", HttpStatus.NOT_FOUND));
 
-    if (entity.getHyperVer().equals("HPs_0001")) {
+    if (entity.getIsDefault()) {
       throw new CustomApiException("UNPROCESSABLE_ENTITY_UPDATE", HttpStatus.UNPROCESSABLE_ENTITY);
     }
     applyHyperParam(entity, createReq);
@@ -69,11 +72,112 @@ public class HyperParamCoreService {
     return entity.getHyperVer();
   }
 
+
+  /**
+   * 하이퍼파라미터 삭제
+   *
+   * @param uuid
+   */
+  public void deleteHyperParam(UUID uuid) {
+    ModelHyperParamEntity entity =
+      hyperParamRepository
+        .findHyperParamByUuid(uuid)
+        .orElseThrow(() -> new CustomApiException("NOT_FOUND_DATA", HttpStatus.NOT_FOUND));
+
+//    if (entity.getHyperVer().equals("HPs_0001")) {
+//      throw new CustomApiException("UNPROCESSABLE_ENTITY", HttpStatus.UNPROCESSABLE_ENTITY);
+//    }
+
+    //디폴트면 삭제불가
+    if (entity.getIsDefault()) {
+      throw new CustomApiException("UNPROCESSABLE_ENTITY", HttpStatus.UNPROCESSABLE_ENTITY);
+    }
+
+    entity.setDelYn(true);
+    entity.setUpdatedUid(userUtil.getId());
+    entity.setUpdatedDttm(ZonedDateTime.now());
+  }
+
+  /**
+   * 하이퍼파라미터 최적화 설정값 조회
+   *
+   * @return
+   */
+  public HyperParamDto.Basic getInitHyperParam(ModelType model) {
+    ModelHyperParamEntity entity =
+      hyperParamRepository
+        .getHyperparamByType(model)
+        .orElseThrow(() -> new CustomApiException("NOT_FOUND_DATA", HttpStatus.NOT_FOUND));
+    return entity.toDto();
+  }
+
+  /**
+   * 하이퍼파라미터 상세 조회
+   *
+   * @return
+   */
+  public HyperParamDto.Basic getHyperParam(UUID uuid) {
+    ModelHyperParamEntity entity =
+      hyperParamRepository
+        .findHyperParamByUuid(uuid)
+        .orElseThrow(() -> new CustomApiException("NOT_FOUND_DATA", HttpStatus.NOT_FOUND));
+    return entity.toDto();
+  }
+
+  /**
+   * 하이퍼파라미터 목록 조회
+   *
+   * @param model
+   * @param req
+   * @return
+   */
+  public Page<HyperParamDto.List> findByHyperVerList(ModelType model, SearchReq req) {
+    return hyperParamRepository.findByHyperVerList(model, req);
+  }
+
+  /**
+   * 하이퍼파라미터 버전 조회
+   *
+   * @param model 모델 타입
+   * @return ver
+   */
+  public String getFirstHyperParamVersion(ModelType model) {
+    return hyperParamRepository
+      .findHyperParamVerByModelType(model)
+      .map(ModelHyperParamEntity::getHyperVer)
+      .map(ver -> increase(ver, model))
+      .orElse(model.name() + "_000001");
+  }
+
+  /**
+   * 하이퍼 파라미터의 버전을 증가시킨다.
+   *
+   * @param hyperVer  현재 버전
+   * @param modelType 모델 타입
+   * @return 증가된 버전
+   */
+  private String increase(String hyperVer, ModelType modelType) {
+    String prefix = modelType.name() + "_";
+    int num = Integer.parseInt(hyperVer.substring(prefix.length()));
+    return prefix + String.format("%06d", num + 1);
+  }
+
   private void applyHyperParam(ModelHyperParamEntity entity, HyperParam src) {
+
+    ModelType model = src.getModel();
+
+    // 하드코딩 모델별로 다른경우 250212 bbn 하드코딩
+    if (model == ModelType.G3) {
+      entity.setCropSize("512,512");
+    } else {
+      entity.setCropSize("256,256");
+    }
+    //    entity.setCropSize(src.getCropSize());
+
     // Important
+    entity.setModelType(model); // 20250212 modeltype추가
     entity.setBackbone(src.getBackbone());
     entity.setInputSize(src.getInputSize());
-    entity.setCropSize(src.getCropSize());
     entity.setBatchSize(src.getBatchSize());
 
     // Data
@@ -111,78 +215,4 @@ public class HyperParamCoreService {
     entity.setMemo(src.getMemo());
   }
 
-  /**
-   * 하이퍼파라미터 삭제
-   *
-   * @param uuid
-   */
-  public void deleteHyperParam(UUID uuid) {
-    ModelHyperParamEntity entity =
-        hyperParamRepository
-            .findHyperParamByUuid(uuid)
-            .orElseThrow(() -> new CustomApiException("NOT_FOUND_DATA", HttpStatus.NOT_FOUND));
-
-    if (entity.getHyperVer().equals("HPs_0001")) {
-      throw new CustomApiException("UNPROCESSABLE_ENTITY", HttpStatus.UNPROCESSABLE_ENTITY);
-    }
-
-    entity.setDelYn(true);
-    entity.setUpdatedUid(userUtil.getId());
-    entity.setUpdatedDttm(ZonedDateTime.now());
-  }
-
-  /**
-   * 하이퍼파라미터 최적화 설정값 조회
-   *
-   * @return
-   */
-  public HyperParamDto.Basic getInitHyperParam() {
-    ModelHyperParamEntity entity =
-        hyperParamRepository
-            .findHyperParamByHyperVer("HPs_0001")
-            .orElseThrow(() -> new CustomApiException("NOT_FOUND_DATA", HttpStatus.NOT_FOUND));
-    return entity.toDto();
-  }
-
-  /**
-   * 하이퍼파라미터 상세 조회
-   *
-   * @return
-   */
-  public HyperParamDto.Basic getHyperParam(UUID uuid) {
-    ModelHyperParamEntity entity =
-        hyperParamRepository
-            .findHyperParamByUuid(uuid)
-            .orElseThrow(() -> new CustomApiException("NOT_FOUND_DATA", HttpStatus.NOT_FOUND));
-    return entity.toDto();
-  }
-
-  /**
-   * 하이퍼파라미터 목록 조회
-   *
-   * @param req
-   * @return
-   */
-  public Page<HyperParamDto.List> findByHyperVerList(HyperParamDto.SearchReq req) {
-    return hyperParamRepository.findByHyperVerList(req);
-  }
-
-  /**
-   * 하이퍼파라미터 버전 조회
-   *
-   * @return ver
-   */
-  public String getFirstHyperParamVersion() {
-    return hyperParamRepository
-        .findHyperParamVer()
-        .map(ModelHyperParamEntity::getHyperVer)
-        .map(this::increase)
-        .orElse("HPs_0001");
-  }
-
-  private String increase(String hyperVer) {
-    String prefix = "HPs_";
-    int num = Integer.parseInt(hyperVer.substring(prefix.length()));
-    return prefix + String.format("%04d", num + 1);
-  }
 }

src/main/java/com/kamco/cd/training/postgres/core/ModelTestMetricsJobCoreService.java

@@ -1,6 +1,7 @@
 package com.kamco.cd.training.postgres.core;
 
 import com.kamco.cd.training.postgres.repository.train.ModelTestMetricsJobRepository;
+import com.kamco.cd.training.train.dto.ModelTrainMetricsDto.ResponsePathDto;
 import java.util.List;
 import lombok.RequiredArgsConstructor;
 import org.springframework.stereotype.Service;
@@ -18,7 +19,7 @@ public class ModelTestMetricsJobCoreService {
   }
 
   // Test 로직 시작
-  public List<Long> getTestMetricSaveNotYetModelIds() {
+  public List<ResponsePathDto> getTestMetricSaveNotYetModelIds() {
     return modelTestMetricsJobRepository.getTestMetricSaveNotYetModelIds();
   }
 

src/main/java/com/kamco/cd/training/postgres/core/ModelTrainMetricsJobCoreService.java

@@ -1,6 +1,7 @@
 package com.kamco.cd.training.postgres.core;
 
 import com.kamco.cd.training.postgres.repository.train.ModelTrainMetricsJobRepository;
+import com.kamco.cd.training.train.dto.ModelTrainMetricsDto.ResponsePathDto;
 import java.util.List;
 import lombok.RequiredArgsConstructor;
 import org.springframework.stereotype.Service;
@@ -12,7 +13,7 @@ public class ModelTrainMetricsJobCoreService {
 
   private final ModelTrainMetricsJobRepository modelTrainMetricsJobRepository;
 
-  public List<Long> getTrainMetricSaveNotYetModelIds() {
+  public List<ResponsePathDto> getTrainMetricSaveNotYetModelIds() {
     return modelTrainMetricsJobRepository.getTrainMetricSaveNotYetModelIds();
   }
 

src/main/java/com/kamco/cd/training/postgres/core/ModelTrainMngCoreService.java

@@ -25,6 +25,7 @@ import com.kamco.cd.training.postgres.repository.model.ModelDatasetRepository;
 import com.kamco.cd.training.postgres.repository.model.ModelMngRepository;
 import com.kamco.cd.training.train.dto.TrainRunRequest;
 import java.time.ZonedDateTime;
+import java.util.ArrayList;
 import java.util.List;
 import java.util.UUID;
 import lombok.RequiredArgsConstructor;
@@ -64,9 +65,9 @@ public class ModelTrainMngCoreService {
    */
   public void deleteModel(UUID uuid) {
     ModelMasterEntity entity =
-        modelMngRepository
-            .findByUuid(uuid)
-            .orElseThrow(() -> new CustomApiException("NOT_FOUND_DATA", HttpStatus.NOT_FOUND));
+      modelMngRepository
+        .findByUuid(uuid)
+        .orElseThrow(() -> new CustomApiException("NOT_FOUND_DATA", HttpStatus.NOT_FOUND));
     entity.setDelYn(true);
     entity.setUpdatedDttm(ZonedDateTime.now());
     entity.setUpdatedUid(userUtil.getId());
@@ -78,17 +79,19 @@ public class ModelTrainMngCoreService {
    * @param addReq
    * @return
    */
-  public Long saveModel(ModelTrainMngDto.AddReq addReq) {
+  public ModelTrainMngDto.Basic saveModel(ModelTrainMngDto.AddReq addReq) {
     ModelMasterEntity entity = new ModelMasterEntity();
     ModelHyperParamEntity hyperParamEntity = new ModelHyperParamEntity();
 
-    // 최적화 파라미터는 HPs_0001 사용
+//     최적화 파라미터는 모델 type의 디폴트사용
     if (HyperParamSelectType.OPTIMIZED.getId().equals(addReq.getHyperParamType())) {
-      hyperParamEntity = hyperParamRepository.findByHyperVer("HPs_0001").orElse(null);
+      ModelType modelType = ModelType.getValueData(addReq.getModelNo());
+      hyperParamEntity = hyperParamRepository.getHyperparamByType(modelType).orElse(null);
+//      hyperParamEntity = hyperParamRepository.findByHyperVer("HPs_0001").orElse(null);
 
     } else {
       hyperParamEntity =
-          hyperParamRepository.findHyperParamByUuid(addReq.getHyperUuid()).orElse(null);
+        hyperParamRepository.findHyperParamByUuid(addReq.getHyperUuid()).orElse(null);
     }
 
     if (hyperParamEntity == null || hyperParamEntity.getHyperVer() == null) {
@@ -96,8 +99,8 @@ public class ModelTrainMngCoreService {
     }
 
     String modelVer =
-        String.join(
-            ".", addReq.getModelNo(), hyperParamEntity.getHyperVer(), entity.getUuid().toString());
+      String.join(
+        ".", addReq.getModelNo(), hyperParamEntity.getHyperVer(), entity.getUuid().toString());
     entity.setModelVer(modelVer);
     entity.setHyperParamId(hyperParamEntity.getId());
     entity.setModelNo(addReq.getModelNo());
@@ -117,14 +120,19 @@ public class ModelTrainMngCoreService {
 
     entity.setCreatedUid(userUtil.getId());
     ModelMasterEntity resultEntity = modelMngRepository.save(entity);
-    return resultEntity.getId();
+
+    ModelTrainMngDto.Basic result = new ModelTrainMngDto.Basic();
+    result.setId(resultEntity.getId());
+    result.setUuid(resultEntity.getUuid());
+
+    return result;
   }
 
   /**
    * data set 저장
    *
    * @param modelId 저장한 모델 학습 id
-   * @param addReq 요청 파라미터
+   * @param addReq  요청 파라미터
    */
   public void saveModelDataset(Long modelId, ModelTrainMngDto.AddReq addReq) {
     TrainingDataset dataset = addReq.getTrainingDataset();
@@ -149,10 +157,30 @@ public class ModelTrainMngCoreService {
     modelDatasetRepository.save(datasetEntity);
   }
 
+  /**
+   * 학습모델 수정
+   *
+   * @param modelId
+   * @param req
+   */
+  public void updateModelMaster(Long modelId, ModelTrainMngDto.UpdateReq req) {
+    ModelMasterEntity entity =
+      modelMngRepository
+        .findById(modelId)
+        .orElseThrow(() -> new CustomApiException("NOT_FOUND_DATA", HttpStatus.NOT_FOUND));
+    if (req.getRequestPath() != null && !req.getRequestPath().isEmpty()) {
+      entity.setRequestPath(req.getRequestPath());
+    }
+
+    if (req.getResponsePath() != null && !req.getResponsePath().isEmpty()) {
+      entity.setRequestPath(req.getResponsePath());
+    }
+  }
+
   /**
    * 모델 데이터셋 mapping 테이블 저장
    *
-   * @param modelId 모델학습 id
+   * @param modelId     모델학습 id
    * @param datasetList 선택한 data set
    */
   public void saveModelDatasetMap(Long modelId, List<Long> datasetList) {
@@ -169,7 +197,7 @@ public class ModelTrainMngCoreService {
    * 모델학습 config 저장
    *
    * @param modelId 모델학습 id
-   * @param req 요청 파라미터
+   * @param req     요청 파라미터
    * @return
    */
   public void saveModelConfig(Long modelId, ModelTrainMngDto.ModelConfig req) {
@@ -189,7 +217,7 @@ public class ModelTrainMngCoreService {
   /**
    * 데이터셋 매핑 생성
    *
-   * @param modelUid 모델 UID
+   * @param modelUid   모델 UID
    * @param datasetIds 데이터셋 ID 목록
    */
   public void createDatasetMappings(Long modelUid, List<Long> datasetIds) {
@@ -211,8 +239,8 @@ public class ModelTrainMngCoreService {
   public ModelMasterEntity findByUuid(UUID uuid) {
     try {
       return modelMngRepository
-          .findByUuid(uuid)
-          .orElseThrow(() -> new CustomApiException("NOT_FOUND_DATA", HttpStatus.NOT_FOUND));
+        .findByUuid(uuid)
+        .orElseThrow(() -> new CustomApiException("NOT_FOUND_DATA", HttpStatus.NOT_FOUND));
     } catch (IllegalArgumentException e) {
       throw new BadRequestException("잘못된 UUID 형식입니다: " + uuid);
     }
@@ -226,9 +254,9 @@ public class ModelTrainMngCoreService {
    */
   public Long findModelIdByUuid(UUID uuid) {
     ModelMasterEntity entity =
-        modelMngRepository
-            .findByUuid(uuid)
-            .orElseThrow(() -> new CustomApiException("NOT_FOUND_DATA", HttpStatus.NOT_FOUND));
+      modelMngRepository
+        .findByUuid(uuid)
+        .orElseThrow(() -> new CustomApiException("NOT_FOUND_DATA", HttpStatus.NOT_FOUND));
     return entity.getId();
   }
 
@@ -241,8 +269,8 @@ public class ModelTrainMngCoreService {
   public ModelConfigDto.Basic findModelConfigByModelId(UUID uuid) {
     ModelMasterEntity modelEntity = findByUuid(uuid);
     return modelConfigRepository
-        .findModelConfigByModelId(modelEntity.getId())
-        .orElseThrow(() -> new CustomApiException("NOT_FOUND_DATA", HttpStatus.NOT_FOUND));
+      .findModelConfigByModelId(modelEntity.getId())
+      .orElseThrow(() -> new CustomApiException("NOT_FOUND_DATA", HttpStatus.NOT_FOUND));
   }
 
   /**
@@ -273,19 +301,21 @@ public class ModelTrainMngCoreService {
    */
   public ModelTrainMngDto.Basic findModelById(Long id) {
     ModelMasterEntity entity =
-        modelMngRepository
-            .findById(id)
-            .orElseThrow(() -> new IllegalArgumentException("Model not found: " + id));
+      modelMngRepository
+        .findById(id)
+        .orElseThrow(() -> new IllegalArgumentException("Model not found: " + id));
     return entity.toDto();
   }
 
-  /** 마스터를 IN_PROGRESS로 전환하고, 현재 실행 jobId를 연결 - UI/중단/상태조회 모두 currentAttemptId를 기준으로 동작 */
+  /**
+   * 마스터를 IN_PROGRESS로 전환하고, 현재 실행 jobId를 연결 - UI/중단/상태조회 모두 currentAttemptId를 기준으로 동작
+   */
   @Transactional
   public void markInProgress(Long modelId, Long jobId) {
     ModelMasterEntity master =
-        modelMngRepository
-            .findById(modelId)
-            .orElseThrow(() -> new IllegalArgumentException("Model not found: " + modelId));
+      modelMngRepository
+        .findById(modelId)
+        .orElseThrow(() -> new IllegalArgumentException("Model not found: " + modelId));
 
     master.setStatusCd(TrainStatusType.IN_PROGRESS.getId());
     master.setCurrentAttemptId(jobId);
@@ -293,46 +323,54 @@ public class ModelTrainMngCoreService {
     // 필요하면 시작시간도 여기서 찍어줌
   }
 
-  /** 마지막 에러 메시지 초기화 - 재시작/새 실행 때 이전 에러 흔적 제거 */
+  /**
+   * 마지막 에러 메시지 초기화 - 재시작/새 실행 때 이전 에러 흔적 제거
+   */
   @Transactional
   public void clearLastError(Long modelId) {
     ModelMasterEntity master =
-        modelMngRepository
-            .findById(modelId)
-            .orElseThrow(() -> new IllegalArgumentException("Model not found: " + modelId));
+      modelMngRepository
+        .findById(modelId)
+        .orElseThrow(() -> new IllegalArgumentException("Model not found: " + modelId));
 
     master.setLastError(null);
   }
 
-  /** 중단 처리(옵션) - cancel에서 쓰려고 하면 같이 구현 */
+  /**
+   * 중단 처리(옵션) - cancel에서 쓰려고 하면 같이 구현
+   */
   @Transactional
   public void markStopped(Long modelId) {
     ModelMasterEntity master =
-        modelMngRepository
-            .findById(modelId)
-            .orElseThrow(() -> new IllegalArgumentException("Model not found: " + modelId));
+      modelMngRepository
+        .findById(modelId)
+        .orElseThrow(() -> new IllegalArgumentException("Model not found: " + modelId));
 
     master.setStatusCd(TrainStatusType.STOPPED.getId());
   }
 
-  /** 완료 처리(옵션) - Worker가 성공 시 호출 */
+  /**
+   * 완료 처리(옵션) - Worker가 성공 시 호출
+   */
   @Transactional
   public void markCompleted(Long modelId) {
     ModelMasterEntity master =
-        modelMngRepository
-            .findById(modelId)
-            .orElseThrow(() -> new IllegalArgumentException("Model not found: " + modelId));
+      modelMngRepository
+        .findById(modelId)
+        .orElseThrow(() -> new IllegalArgumentException("Model not found: " + modelId));
 
     master.setStatusCd(TrainStatusType.COMPLETED.getId());
   }
 
-  /** step 1오류 처리(옵션) - Worker가 실패 시 호출 */
+  /**
+   * step 1오류 처리(옵션) - Worker가 실패 시 호출
+   */
   @Transactional
   public void markError(Long modelId, String errorMessage) {
     ModelMasterEntity master =
-        modelMngRepository
-            .findById(modelId)
-            .orElseThrow(() -> new IllegalArgumentException("Model not found: " + modelId));
+      modelMngRepository
+        .findById(modelId)
+        .orElseThrow(() -> new IllegalArgumentException("Model not found: " + modelId));
 
     master.setStatusCd(TrainStatusType.ERROR.getId());
     master.setStep1State(TrainStatusType.ERROR.getId());
@@ -341,13 +379,15 @@ public class ModelTrainMngCoreService {
     master.setUpdatedDttm(ZonedDateTime.now());
   }
 
-  /** step 2오류 처리(옵션) - Worker가 실패 시 호출 */
+  /**
+   * step 2오류 처리(옵션) - Worker가 실패 시 호출
+   */
   @Transactional
   public void markStep2Error(Long modelId, String errorMessage) {
     ModelMasterEntity master =
-        modelMngRepository
-            .findById(modelId)
-            .orElseThrow(() -> new IllegalArgumentException("Model not found: " + modelId));
+      modelMngRepository
+        .findById(modelId)
+        .orElseThrow(() -> new IllegalArgumentException("Model not found: " + modelId));
 
     master.setStatusCd(TrainStatusType.ERROR.getId());
     master.setStep2State(TrainStatusType.ERROR.getId());
@@ -359,9 +399,9 @@ public class ModelTrainMngCoreService {
   @Transactional
   public void markSuccess(Long modelId) {
     ModelMasterEntity master =
-        modelMngRepository
-            .findById(modelId)
-            .orElseThrow(() -> new IllegalArgumentException("Model not found: " + modelId));
+      modelMngRepository
+        .findById(modelId)
+        .orElseThrow(() -> new IllegalArgumentException("Model not found: " + modelId));
 
     // 모델 상태 완료 처리
     master.setStatusCd(TrainStatusType.COMPLETED.getId());
@@ -389,9 +429,9 @@ public class ModelTrainMngCoreService {
   @Transactional(propagation = Propagation.REQUIRES_NEW)
   public void markStep1InProgress(Long modelId, Long jobId) {
     ModelMasterEntity entity =
-        modelMngRepository
-            .findById(modelId)
-            .orElseThrow(() -> new IllegalArgumentException("Model not found: " + modelId));
+      modelMngRepository
+        .findById(modelId)
+        .orElseThrow(() -> new IllegalArgumentException("Model not found: " + modelId));
 
     entity.setStatusCd(TrainStatusType.IN_PROGRESS.getId());
     entity.setStep1StrtDttm(ZonedDateTime.now());
@@ -409,9 +449,9 @@ public class ModelTrainMngCoreService {
   @Transactional(propagation = Propagation.REQUIRES_NEW)
   public void markStep2InProgress(Long modelId, Long jobId) {
     ModelMasterEntity entity =
-        modelMngRepository
-            .findById(modelId)
-            .orElseThrow(() -> new IllegalArgumentException("Model not found: " + modelId));
+      modelMngRepository
+        .findById(modelId)
+        .orElseThrow(() -> new IllegalArgumentException("Model not found: " + modelId));
 
     entity.setStatusCd(TrainStatusType.IN_PROGRESS.getId());
     entity.setStep2StrtDttm(ZonedDateTime.now());
@@ -429,9 +469,9 @@ public class ModelTrainMngCoreService {
   @Transactional(propagation = Propagation.REQUIRES_NEW)
   public void markStep1Success(Long modelId) {
     ModelMasterEntity entity =
-        modelMngRepository
-            .findById(modelId)
-            .orElseThrow(() -> new IllegalArgumentException("Model not found: " + modelId));
+      modelMngRepository
+        .findById(modelId)
+        .orElseThrow(() -> new IllegalArgumentException("Model not found: " + modelId));
 
     entity.setStatusCd(TrainStatusType.COMPLETED.getId());
     entity.setStep1State(TrainStatusType.COMPLETED.getId());
@@ -448,9 +488,9 @@ public class ModelTrainMngCoreService {
   @Transactional(propagation = Propagation.REQUIRES_NEW)
   public void markStep2Success(Long modelId) {
     ModelMasterEntity entity =
-        modelMngRepository
-            .findById(modelId)
-            .orElseThrow(() -> new IllegalArgumentException("Model not found: " + modelId));
+      modelMngRepository
+        .findById(modelId)
+        .orElseThrow(() -> new IllegalArgumentException("Model not found: " + modelId));
 
     entity.setStatusCd(TrainStatusType.COMPLETED.getId());
     entity.setStep2State(TrainStatusType.COMPLETED.getId());
@@ -461,10 +501,34 @@ public class ModelTrainMngCoreService {
 
   public void updateModelMasterBestEpoch(Long modelId, int epoch) {
     ModelMasterEntity entity =
-        modelMngRepository
-            .findById(modelId)
-            .orElseThrow(() -> new IllegalArgumentException("Model not found: " + modelId));
+      modelMngRepository
+        .findById(modelId)
+        .orElseThrow(() -> new IllegalArgumentException("Model not found: " + modelId));
 
     entity.setBestEpoch(epoch);
   }
+
+  /**
+   * 데이터셋 uid 조회
+   *
+   * @param datasetIds
+   * @return
+   */
+  public List<String> findDatasetUid(List<Long> datasetIds) {
+    return datasetRepository.findDatasetUid(datasetIds);
+  }
+
+  public List<Long> findModelDatasetMapp(Long modelId) {
+    List<Long> datasetUids = new ArrayList<>();
+    List<ModelDatasetMappEntity> entities = modelDatasetMapRepository.findByModelUid(modelId);
+    for (ModelDatasetMappEntity entity : entities) {
+      datasetUids.add(entity.getDatasetUid());
+    }
+
+    return datasetUids;
+  }
+
+  public Long findModelStep1InProgressCnt() {
+    return modelMngRepository.findModelStep1InProgressCnt();
+  }
 }

src/main/java/com/kamco/cd/training/postgres/entity/ModelHyperParamEntity.java

@@ -1,5 +1,6 @@
 package com.kamco.cd.training.postgres.entity;
 
+import com.kamco.cd.training.common.enums.ModelType;
 import com.kamco.cd.training.hyperparam.dto.HyperParamDto;
 import jakarta.persistence.*;
 import jakarta.validation.constraints.NotNull;
@@ -311,8 +312,17 @@ public class ModelHyperParamEntity {
   @Column(name = "m3_use_cnt")
   private Long m3UseCnt = 0L;
 
+  @Column(name = "model_type")
+  @Enumerated(EnumType.STRING)
+  private ModelType modelType;
+
+
+  @Column(name = "default_param")
+  private Boolean isDefault = false;
+
   public HyperParamDto.Basic toDto() {
     return new HyperParamDto.Basic(
+        this.modelType,
         this.uuid,
         this.hyperVer,
         this.createdDttm,
@@ -385,6 +395,8 @@ public class ModelHyperParamEntity {
         // -------------------------
         this.gpuCnt,
         this.gpuIds,
-        this.masterPort);
+        this.masterPort
+      , this.isDefault
+    );
   }
 }

src/main/java/com/kamco/cd/training/postgres/entity/ModelMasterEntity.java

@@ -106,6 +106,12 @@ public class ModelMasterEntity {
   @Column(name = "best_epoch")
   private Integer bestEpoch;
 
+  @Column(name = "request_path")
+  private String requestPath;
+
+  @Column(name = "response_path")
+  private String responsePath;
+
   public ModelTrainMngDto.Basic toDto() {
     return new ModelTrainMngDto.Basic(
         this.id,

src/main/java/com/kamco/cd/training/postgres/repository/dataset/DatasetRepositoryCustom.java

@@ -22,4 +22,6 @@ public interface DatasetRepositoryCustom {
   Long getDatasetMaxStage(int compareYyyy, int targetYyyy);
 
   Long insertDatasetMngData(DatasetMngRegDto mngRegDto);
+
+  List<String> findDatasetUid(List<Long> datasetIds);
 }

src/main/java/com/kamco/cd/training/postgres/repository/dataset/DatasetRepositoryImpl.java

@@ -242,4 +242,9 @@ public class DatasetRepositoryImpl implements DatasetRepositoryCustom {
         .where(dataset.uid.eq(mngRegDto.getUid()))
         .fetchOne();
   }
+
+  @Override
+  public List<String> findDatasetUid(List<Long> datasetIds) {
+    return queryFactory.select(dataset.uid).from(dataset).where(dataset.id.in(datasetIds)).fetch();
+  }
 }

src/main/java/com/kamco/cd/training/postgres/repository/hyperparam/HyperParamRepositoryCustom.java

@@ -1,6 +1,8 @@
 package com.kamco.cd.training.postgres.repository.hyperparam;
 
+import com.kamco.cd.training.common.enums.ModelType;
 import com.kamco.cd.training.hyperparam.dto.HyperParamDto;
+import com.kamco.cd.training.hyperparam.dto.HyperParamDto.SearchReq;
 import com.kamco.cd.training.postgres.entity.ModelHyperParamEntity;
 import java.util.Optional;
 import java.util.UUID;
@@ -13,11 +15,22 @@ public interface HyperParamRepositoryCustom {
    *
    * @return
    */
+  @Deprecated
   Optional<ModelHyperParamEntity> findHyperParamVer();
 
+  /**
+   * 모델 타입별 마지막 버전 조회
+   *
+   * @param modelType 모델 타입
+   * @return
+   */
+  Optional<ModelHyperParamEntity> findHyperParamVerByModelType(ModelType modelType);
+
   Optional<ModelHyperParamEntity> findHyperParamByHyperVer(String hyperVer);
 
   Optional<ModelHyperParamEntity> findHyperParamByUuid(UUID uuid);
 
-  Page<HyperParamDto.List> findByHyperVerList(HyperParamDto.SearchReq req);
+  Page<HyperParamDto.List> findByHyperVerList(ModelType model, SearchReq req);
+
+  Optional<ModelHyperParamEntity> getHyperparamByType(ModelType modelType);
 }

src/main/java/com/kamco/cd/training/postgres/repository/hyperparam/HyperParamRepositoryImpl.java

@@ -2,8 +2,10 @@ package com.kamco.cd.training.postgres.repository.hyperparam;
 
 import static com.kamco.cd.training.postgres.entity.QModelHyperParamEntity.modelHyperParamEntity;
 
+import com.kamco.cd.training.common.enums.ModelType;
 import com.kamco.cd.training.hyperparam.dto.HyperParamDto;
 import com.kamco.cd.training.hyperparam.dto.HyperParamDto.HyperType;
+import com.kamco.cd.training.hyperparam.dto.HyperParamDto.SearchReq;
 import com.kamco.cd.training.postgres.entity.ModelHyperParamEntity;
 import com.querydsl.core.BooleanBuilder;
 import com.querydsl.core.types.Projections;
@@ -41,6 +43,23 @@ public class HyperParamRepositoryImpl implements HyperParamRepositoryCustom {
             .fetchOne());
   }
 
+  @Override
+  public Optional<ModelHyperParamEntity> findHyperParamVerByModelType(ModelType modelType) {
+
+    return Optional.ofNullable(
+        queryFactory
+            .select(modelHyperParamEntity)
+            .from(modelHyperParamEntity)
+            .where(
+                modelHyperParamEntity
+                    .delYn
+                    .isFalse()
+                    .and(modelHyperParamEntity.modelType.eq(modelType)))
+            .orderBy(modelHyperParamEntity.hyperVer.desc())
+            .limit(1)
+            .fetchOne());
+  }
+
   @Override
   public Optional<ModelHyperParamEntity> findHyperParamByHyperVer(String hyperVer) {
 
@@ -68,10 +87,11 @@ public class HyperParamRepositoryImpl implements HyperParamRepositoryCustom {
   }
 
   @Override
-  public Page<HyperParamDto.List> findByHyperVerList(HyperParamDto.SearchReq req) {
+  public Page<HyperParamDto.List> findByHyperVerList(ModelType model, SearchReq req) {
     Pageable pageable = req.toPageable();
 
     BooleanBuilder builder = new BooleanBuilder();
+    builder.and(modelHyperParamEntity.modelType.eq(model));
     builder.and(modelHyperParamEntity.delYn.isFalse());
 
     if (req.getHyperVer() != null && !req.getHyperVer().isEmpty()) {
@@ -161,4 +181,14 @@ public class HyperParamRepositoryImpl implements HyperParamRepositoryCustom {
 
     return new PageImpl<>(content, pageable, totalCount);
   }
+
+  @Override
+  public Optional<ModelHyperParamEntity> getHyperparamByType(ModelType modelType) {
+    return Optional.ofNullable(
+      queryFactory
+        .select(modelHyperParamEntity)
+        .from(modelHyperParamEntity)
+        .where(modelHyperParamEntity.delYn.isFalse().and(modelHyperParamEntity.modelType.eq(modelType)))
+        .fetchOne());
+  }
 }

src/main/java/com/kamco/cd/training/postgres/repository/model/ModelDatasetMappRepository.java

@@ -6,4 +6,5 @@ import org.springframework.stereotype.Repository;
 
 @Repository
 public interface ModelDatasetMappRepository
-    extends JpaRepository<ModelDatasetMappEntity, ModelDatasetMappEntity.ModelDatasetMappId> {}
+    extends JpaRepository<ModelDatasetMappEntity, ModelDatasetMappEntity.ModelDatasetMappId>,
+        ModelDatasetMappRepositoryCustom {}

src/main/java/com/kamco/cd/training/postgres/repository/model/ModelDatasetMappRepositoryCustom.java

@@ -0,0 +1,8 @@
+package com.kamco.cd.training.postgres.repository.model;
+
+import com.kamco.cd.training.postgres.entity.ModelDatasetMappEntity;
+import java.util.List;
+
+public interface ModelDatasetMappRepositoryCustom {
+  List<ModelDatasetMappEntity> findByModelUid(Long modelId);
+}

src/main/java/com/kamco/cd/training/postgres/repository/model/ModelDatasetMappRepositoryImpl.java

@@ -0,0 +1,25 @@
+package com.kamco.cd.training.postgres.repository.model;
+
+import static com.kamco.cd.training.postgres.entity.QModelDatasetMappEntity.modelDatasetMappEntity;
+
+import com.kamco.cd.training.postgres.entity.ModelDatasetMappEntity;
+import com.querydsl.jpa.impl.JPAQueryFactory;
+import java.util.List;
+import lombok.RequiredArgsConstructor;
+import org.springframework.stereotype.Repository;
+
+@Repository
+@RequiredArgsConstructor
+public class ModelDatasetMappRepositoryImpl implements ModelDatasetMappRepositoryCustom {
+
+  private final JPAQueryFactory queryFactory;
+
+  @Override
+  public List<ModelDatasetMappEntity> findByModelUid(Long modelId) {
+    queryFactory
+        .select(modelDatasetMappEntity)
+        .from(modelDatasetMappEntity)
+        .where(modelDatasetMappEntity.modelUid.eq(modelId));
+    return List.of();
+  }
+}

src/main/java/com/kamco/cd/training/postgres/repository/model/ModelMngRepositoryCustom.java

@@ -22,4 +22,6 @@ public interface ModelMngRepositoryCustom {
   Optional<ModelMasterEntity> findFirstByStatusCdAndDelYn(String statusCd, Boolean delYn);
 
   TrainRunRequest findTrainRunRequest(Long modelId);
+
+  Long findModelStep1InProgressCnt();
 }

src/main/java/com/kamco/cd/training/postgres/repository/model/ModelMngRepositoryImpl.java

@@ -4,6 +4,7 @@ import static com.kamco.cd.training.postgres.entity.QModelConfigEntity.modelConf
 import static com.kamco.cd.training.postgres.entity.QModelHyperParamEntity.modelHyperParamEntity;
 import static com.kamco.cd.training.postgres.entity.QModelMasterEntity.modelMasterEntity;
 
+import com.kamco.cd.training.common.enums.TrainStatusType;
 import com.kamco.cd.training.model.dto.ModelTrainMngDto;
 import com.kamco.cd.training.postgres.entity.ModelMasterEntity;
 import com.kamco.cd.training.train.dto.TrainRunRequest;
@@ -45,6 +46,8 @@ public class ModelMngRepositoryImpl implements ModelMngRepositoryCustom {
       builder.and(modelMasterEntity.modelNo.eq(req.getModelNo()));
     }
 
+    builder.and(modelMasterEntity.delYn.isFalse());
+
     List<ModelMasterEntity> content =
         queryFactory
             .selectFrom(modelMasterEntity)
@@ -94,7 +97,7 @@ public class ModelMngRepositoryImpl implements ModelMngRepositoryCustom {
         .select(
             Projections.constructor(
                 TrainRunRequest.class,
-                modelMasterEntity.uuid, // datasetFolder
+                modelMasterEntity.requestPath, // datasetFolder
                 modelMasterEntity.uuid, // outputFolder
                 modelHyperParamEntity.inputSize,
                 modelHyperParamEntity.cropSize,
@@ -145,4 +148,13 @@ public class ModelMngRepositoryImpl implements ModelMngRepositoryCustom {
         .where(modelMasterEntity.id.eq(modelId))
         .fetchOne();
   }
+
+  @Override
+  public Long findModelStep1InProgressCnt() {
+    return queryFactory
+        .select(modelMasterEntity.id.count())
+        .from(modelMasterEntity)
+        .where(modelMasterEntity.step1State.eq(TrainStatusType.IN_PROGRESS.getId()))
+        .fetchOne();
+  }
 }

src/main/java/com/kamco/cd/training/postgres/repository/train/ModelTestMetricsJobRepositoryCustom.java

@@ -1,12 +1,13 @@
 package com.kamco.cd.training.postgres.repository.train;
 
+import com.kamco.cd.training.train.dto.ModelTrainMetricsDto.ResponsePathDto;
 import java.util.List;
 
 public interface ModelTestMetricsJobRepositoryCustom {
 
   void updateModelMetricsTrainSaveYn(Long modelId, String stepNo);
 
-  List<Long> getTestMetricSaveNotYetModelIds();
+  List<ResponsePathDto> getTestMetricSaveNotYetModelIds();
 
   void insertModelMetricsTest(List<Object[]> batchArgs);
 }

src/main/java/com/kamco/cd/training/postgres/repository/train/ModelTestMetricsJobRepositoryImpl.java

@@ -4,6 +4,8 @@ import static com.kamco.cd.training.postgres.entity.QModelMasterEntity.modelMast
 
 import com.kamco.cd.training.common.enums.TrainStatusType;
 import com.kamco.cd.training.postgres.entity.ModelMetricsTestEntity;
+import com.kamco.cd.training.train.dto.ModelTrainMetricsDto.ResponsePathDto;
+import com.querydsl.core.types.Projections;
 import com.querydsl.jpa.impl.JPAQueryFactory;
 import java.util.List;
 import org.springframework.data.jpa.repository.support.QuerydslRepositorySupport;
@@ -36,9 +38,11 @@ public class ModelTestMetricsJobRepositoryImpl extends QuerydslRepositorySupport
   }
 
   @Override
-  public List<Long> getTestMetricSaveNotYetModelIds() {
+  public List<ResponsePathDto> getTestMetricSaveNotYetModelIds() {
     return queryFactory
-        .select(modelMasterEntity.id)
+        .select(
+            Projections.constructor(
+                ResponsePathDto.class, modelMasterEntity.id, modelMasterEntity.responsePath))
         .from(modelMasterEntity)
         .where(
             modelMasterEntity.step2EndDttm.isNotNull(),

src/main/java/com/kamco/cd/training/postgres/repository/train/ModelTrainMetricsJobRepositoryCustom.java

@@ -1,10 +1,11 @@
 package com.kamco.cd.training.postgres.repository.train;
 
+import com.kamco.cd.training.train.dto.ModelTrainMetricsDto.ResponsePathDto;
 import java.util.List;
 
 public interface ModelTrainMetricsJobRepositoryCustom {
 
-  List<Long> getTrainMetricSaveNotYetModelIds();
+  List<ResponsePathDto> getTrainMetricSaveNotYetModelIds();
 
   void insertModelMetricsTrain(List<Object[]> batchArgs);
 

src/main/java/com/kamco/cd/training/postgres/repository/train/ModelTrainMetricsJobRepositoryImpl.java

@@ -4,6 +4,8 @@ import static com.kamco.cd.training.postgres.entity.QModelMasterEntity.modelMast
 
 import com.kamco.cd.training.common.enums.TrainStatusType;
 import com.kamco.cd.training.postgres.entity.ModelMetricsTrainEntity;
+import com.kamco.cd.training.train.dto.ModelTrainMetricsDto.ResponsePathDto;
+import com.querydsl.core.types.Projections;
 import com.querydsl.jpa.impl.JPAQueryFactory;
 import java.util.List;
 import org.springframework.data.jpa.repository.support.QuerydslRepositorySupport;
@@ -23,9 +25,11 @@ public class ModelTrainMetricsJobRepositoryImpl extends QuerydslRepositorySuppor
   }
 
   @Override
-  public List<Long> getTrainMetricSaveNotYetModelIds() {
+  public List<ResponsePathDto> getTrainMetricSaveNotYetModelIds() {
     return queryFactory
-        .select(modelMasterEntity.id)
+        .select(
+            Projections.constructor(
+                ResponsePathDto.class, modelMasterEntity.id, modelMasterEntity.responsePath))
         .from(modelMasterEntity)
         .where(
             modelMasterEntity.step1EndDttm.isNotNull(),
@@ -41,7 +45,7 @@ public class ModelTrainMetricsJobRepositoryImpl extends QuerydslRepositorySuppor
   public void insertModelMetricsTrain(List<Object[]> batchArgs) {
     String sql =
         """
-            insert into tb_model_matrics_train
+            insert into tb_model_metrics_train
             (model_id, epoch, iteration, loss, lr, duration_time)
             values (?, ?, ?, ?, ?, ?)
         """;
@@ -66,7 +70,7 @@ public class ModelTrainMetricsJobRepositoryImpl extends QuerydslRepositorySuppor
   public void insertModelMetricsValidation(List<Object[]> batchArgs) {
     String sql =
         """
-            insert into tb_model_matrics_validation
+            insert into tb_model_metrics_validation
             (model_id, epoch, a_acc, m_fscore, m_precision, m_recall, m_iou, m_acc, changed_fscore, changed_precision, changed_recall,
              unchanged_fscore, unchanged_precision, unchanged_recall
              )

src/main/java/com/kamco/cd/training/train/TrainApiController.java

@@ -164,4 +164,26 @@ public class TrainApiController {
     testJobService.cancel(modelId);
     return ApiResponseDto.ok("ok");
   }
+
+  @Operation(summary = "데이터셋 tmp 파일생성", description = "데이터셋 tmp 파일생성 API")
+  @ApiResponses(
+      value = {
+        @ApiResponse(
+            responseCode = "200",
+            description = "데이터셋 tmp 파일생성 성공",
+            content =
+                @Content(
+                    mediaType = "application/json",
+                    schema = @Schema(implementation = String.class))),
+        @ApiResponse(responseCode = "400", description = "잘못된 검색 조건", content = @Content),
+        @ApiResponse(responseCode = "500", description = "서버 오류", content = @Content)
+      })
+  @PostMapping("/create-tmp/{uuid}")
+  public ApiResponseDto<UUID> createTmpFile(
+      @Parameter(description = "uuid", example = "80a0e544-36ed-4999-b705-97427f23337d")
+          @PathVariable
+          UUID uuid) {
+
+    return ApiResponseDto.ok(trainJobService.createTmpFile(uuid));
+  }
 }

src/main/java/com/kamco/cd/training/train/dto/ModelTrainMetricsDto.java

@@ -0,0 +1,21 @@
+package com.kamco.cd.training.train.dto;
+
+import io.swagger.v3.oas.annotations.media.Schema;
+import lombok.AllArgsConstructor;
+import lombok.Getter;
+import lombok.NoArgsConstructor;
+import lombok.Setter;
+
+public class ModelTrainMetricsDto {
+
+  @Schema(name = "ResponsePathDto", description = "AI 결과 저장된 path 경로 정보")
+  @Getter
+  @Setter
+  @NoArgsConstructor
+  @AllArgsConstructor
+  public static class ResponsePathDto {
+
+    private Long modelId;
+    private String responsePath;
+  }
+}

src/main/java/com/kamco/cd/training/train/dto/TrainRunRequest.java

@@ -15,7 +15,7 @@ public class TrainRunRequest {
   // ========================
   // 기본
   // ========================
-  private UUID datasetFolder;
+  private String datasetFolder;
   private UUID outputFolder;
   private String inputSize;
   private String cropSize;
@@ -84,10 +84,6 @@ public class TrainRunRequest {
 
   private UUID uuid;
 
-  public String getDatasetFolder() {
-    return String.valueOf(this.datasetFolder);
-  }
-
   public String getOutputFolder() {
     return String.valueOf(this.outputFolder);
   }

src/main/java/com/kamco/cd/training/train/service/DockerTrainService.java

@@ -221,7 +221,7 @@ public class DockerTrainService {
     c.add("/workspace/change-detection-code/train_wrapper.py");
 
     // ===== 기본 파라미터 =====
-    addArg(c, "--dataset-folder", "4BDBBDF99D04477A927CC9EBA760B845" /*req.getDatasetFolder()*/);
+    addArg(c, "--dataset-folder", req.getDatasetFolder());
     addArg(c, "--output-folder", req.getOutputFolder());
     addArg(c, "--input-size", req.getInputSize());
     addArg(c, "--crop-size", req.getCropSize());
@@ -281,8 +281,8 @@ public class DockerTrainService {
     if (value == null) return;
     String s = String.valueOf(value).trim();
     if (s.isEmpty()) return;
-    c.add(key);
-    c.add(s);
+
+    c.add(key + "=" + s);
   }
 
   /** 컨테이너 강제 종료 및 제거 */

src/main/java/com/kamco/cd/training/train/service/ModelTestMetricsJobService.java

@@ -1,6 +1,7 @@
 package com.kamco.cd.training.train.service;
 
 import com.kamco.cd.training.postgres.core.ModelTestMetricsJobCoreService;
+import com.kamco.cd.training.train.dto.ModelTrainMetricsDto.ResponsePathDto;
 import java.io.BufferedReader;
 import java.io.IOException;
 import java.nio.charset.StandardCharsets;
@@ -35,66 +36,67 @@ public class ModelTestMetricsJobService {
     return "local".equalsIgnoreCase(profile);
   }
 
-  //  @Scheduled(cron = "0 0/10 * * * *")
+  //    @Scheduled(cron = "0 * * * * *")
   public void findTestValidMetricCsvFiles() {
-    if (isLocalProfile()) {
-      return;
-    }
+    //    if (isLocalProfile()) {
+    //      return;
+    //    }
 
-    List<Long> modelIds =
-        modelTestMetricsJobCoreService
-            .getTestMetricSaveNotYetModelIds(); // TODO: uid, uuid ? 가져오기로 해야함
+    List<ResponsePathDto> modelIds =
+        modelTestMetricsJobCoreService.getTestMetricSaveNotYetModelIds();
 
     if (modelIds.isEmpty()) {
       return;
     }
 
-    String localPath = "C:\\data\\upload\\test.csv";
-    try (BufferedReader reader =
-        Files.newBufferedReader(Paths.get(localPath), StandardCharsets.UTF_8); ) {
+    for (ResponsePathDto modelInfo : modelIds) {
 
-      log.info("### localPath={}", localPath);
-      CSVParser parser = CSVFormat.DEFAULT.withFirstRecordAsHeader().parse(reader);
+      String testPath = modelInfo.getResponsePath() + "/metrics/test.csv";
+      try (BufferedReader reader =
+          Files.newBufferedReader(Paths.get(testPath), StandardCharsets.UTF_8); ) {
 
-      List<Object[]> batchArgs = new ArrayList<>();
+        CSVParser parser = CSVFormat.DEFAULT.withFirstRecordAsHeader().parse(reader);
 
-      for (CSVRecord record : parser) {
+        List<Object[]> batchArgs = new ArrayList<>();
 
-        String model = record.get("model");
-        long TP = Long.parseLong(record.get("TP"));
-        long FP = Long.parseLong(record.get("FP"));
-        long FN = Long.parseLong(record.get("FN"));
-        float precision = Float.parseFloat(record.get("precision"));
-        float recall = Float.parseFloat(record.get("recall"));
-        float f1_score = Float.parseFloat(record.get("f1_score"));
-        float accuracy = Float.parseFloat(record.get("accuracy"));
-        float iou = Float.parseFloat(record.get("iou"));
-        long detection_count = Long.parseLong(record.get("detection_count"));
-        long gt_count = Long.parseLong(record.get("gt_count"));
+        for (CSVRecord record : parser) {
 
-        batchArgs.add(
-            new Object[] {
-              modelIds.getFirst(),
-              model,
-              TP,
-              FP,
-              FN,
-              precision,
-              recall,
-              f1_score,
-              accuracy,
-              iou,
-              detection_count,
-              gt_count
-            });
+          String model = record.get("model");
+          long TP = Long.parseLong(record.get("TP"));
+          long FP = Long.parseLong(record.get("FP"));
+          long FN = Long.parseLong(record.get("FN"));
+          float precision = Float.parseFloat(record.get("precision"));
+          float recall = Float.parseFloat(record.get("recall"));
+          float f1_score = Float.parseFloat(record.get("f1_score"));
+          float accuracy = Float.parseFloat(record.get("accuracy"));
+          float iou = Float.parseFloat(record.get("iou"));
+          long detection_count = Long.parseLong(record.get("detection_count"));
+          long gt_count = Long.parseLong(record.get("gt_count"));
+
+          batchArgs.add(
+              new Object[] {
+                modelInfo.getModelId(),
+                model,
+                TP,
+                FP,
+                FN,
+                precision,
+                recall,
+                f1_score,
+                accuracy,
+                iou,
+                detection_count,
+                gt_count
+              });
+        }
+
+        modelTestMetricsJobCoreService.insertModelMetricsTest(batchArgs);
+
+      } catch (IOException e) {
+        throw new RuntimeException(e);
       }
 
-      modelTestMetricsJobCoreService.insertModelMetricsTest(batchArgs);
-
-    } catch (IOException e) {
-      throw new RuntimeException(e);
+      modelTestMetricsJobCoreService.updateModelMetricsTrainSaveYn(modelInfo.getModelId(), "step2");
     }
-
-    modelTestMetricsJobCoreService.updateModelMetricsTrainSaveYn(modelIds.getFirst(), "step2");
   }
 }

src/main/java/com/kamco/cd/training/train/service/ModelTrainMetricsJobService.java

@@ -1,6 +1,7 @@
 package com.kamco.cd.training.train.service;
 
 import com.kamco.cd.training.postgres.core.ModelTrainMetricsJobCoreService;
+import com.kamco.cd.training.train.dto.ModelTrainMetricsDto.ResponsePathDto;
 import java.io.BufferedReader;
 import java.io.IOException;
 import java.nio.charset.StandardCharsets;
@@ -26,6 +27,10 @@ public class ModelTrainMetricsJobService {
   @Value("${spring.profiles.active}")
   private String profile;
 
+  // 학습 결과가 저장될 호스트 디렉토리
+  @Value("${train.docker.responseDir}")
+  private String responseDir;
+
   /**
    * 실행중인 profile
    *
@@ -35,96 +40,97 @@ public class ModelTrainMetricsJobService {
     return "local".equalsIgnoreCase(profile);
   }
 
-  //    @Scheduled(cron = "0 0/10 * * * *")
+  // @Scheduled(cron = "0 * * * * *")
   public void findTrainValidMetricCsvFiles() {
-    if (isLocalProfile()) {
-      return;
-    }
+    //    if (isLocalProfile()) {
+    //      return;
+    //    }
 
-    List<Long> modelIds =
-        modelTrainMetricsJobCoreService
-            .getTrainMetricSaveNotYetModelIds(); // TODO: uid, uuid ? 가져오기로 해야함
+    List<ResponsePathDto> modelIds =
+        modelTrainMetricsJobCoreService.getTrainMetricSaveNotYetModelIds();
 
     if (modelIds.isEmpty()) {
       return;
     }
 
-    String localPath = "C:\\data\\upload\\train.csv";
-    try (BufferedReader reader =
-        Files.newBufferedReader(Paths.get(localPath), StandardCharsets.UTF_8); ) {
+    for (ResponsePathDto modelInfo : modelIds) {
 
-      log.info("### localPath={}", localPath);
-      CSVParser parser = CSVFormat.DEFAULT.withFirstRecordAsHeader().parse(reader);
+      String trainPath = responseDir + "{uuid}/metrics/train.csv"; // TODO
+      try (BufferedReader reader =
+          Files.newBufferedReader(Paths.get(trainPath), StandardCharsets.UTF_8); ) {
 
-      List<Object[]> batchArgs = new ArrayList<>();
+        CSVParser parser = CSVFormat.DEFAULT.withFirstRecordAsHeader().parse(reader);
 
-      for (CSVRecord record : parser) {
+        List<Object[]> batchArgs = new ArrayList<>();
 
-        int epoch = Integer.parseInt(record.get("Epoch"));
-        long iteration = Long.parseLong(record.get("Iteration"));
-        double Loss = Double.parseDouble(record.get("Loss"));
-        double LR = Double.parseDouble(record.get("LR"));
-        float time = Float.parseFloat(record.get("Time"));
+        for (CSVRecord record : parser) {
 
-        batchArgs.add(new Object[] {modelIds.getFirst(), epoch, iteration, Loss, LR, time});
+          int epoch = Integer.parseInt(record.get("Epoch")) + 1; // TODO : 나중에 AI 개발 완료되면 -1 하기
+          long iteration = Long.parseLong(record.get("Iteration"));
+          double Loss = Double.parseDouble(record.get("Loss"));
+          double LR = Double.parseDouble(record.get("LR"));
+          float time = Float.parseFloat(record.get("Time"));
+
+          batchArgs.add(new Object[] {modelInfo.getModelId(), epoch, iteration, Loss, LR, time});
+        }
+
+        modelTrainMetricsJobCoreService.insertModelMetricsTrain(batchArgs);
+
+      } catch (IOException e) {
+        throw new RuntimeException(e);
       }
 
-      modelTrainMetricsJobCoreService.insertModelMetricsTrain(batchArgs);
+      String validationPath = modelInfo.getResponsePath() + "/metrics/val.csv";
+      try (BufferedReader reader =
+          Files.newBufferedReader(Paths.get(validationPath), StandardCharsets.UTF_8); ) {
 
-    } catch (IOException e) {
-      throw new RuntimeException(e);
-    }
+        CSVParser parser = CSVFormat.DEFAULT.withFirstRecordAsHeader().parse(reader);
 
-    String validationPath = "C:\\data\\upload\\val.csv";
-    try (BufferedReader reader =
-        Files.newBufferedReader(Paths.get(validationPath), StandardCharsets.UTF_8); ) {
+        List<Object[]> batchArgs = new ArrayList<>();
 
-      log.info("### validationPath={}", validationPath);
-      CSVParser parser = CSVFormat.DEFAULT.withFirstRecordAsHeader().parse(reader);
+        for (CSVRecord record : parser) {
 
-      List<Object[]> batchArgs = new ArrayList<>();
+          int epoch = Integer.parseInt(record.get("Epoch"));
+          float aAcc = Float.parseFloat(record.get("aAcc"));
+          float mFscore = Float.parseFloat(record.get("mFscore"));
+          float mPrecision = Float.parseFloat(record.get("mPrecision"));
+          float mRecall = Float.parseFloat(record.get("mRecall"));
+          float mIoU = Float.parseFloat(record.get("mIoU"));
+          float mAcc = Float.parseFloat(record.get("mAcc"));
+          float changed_fscore = Float.parseFloat(record.get("changed_fscore"));
+          float changed_precision = Float.parseFloat(record.get("changed_precision"));
+          float changed_recall = Float.parseFloat(record.get("changed_recall"));
+          float unchanged_fscore = Float.parseFloat(record.get("unchanged_fscore"));
+          float unchanged_precision = Float.parseFloat(record.get("unchanged_precision"));
+          float unchanged_recall = Float.parseFloat(record.get("unchanged_recall"));
 
-      for (CSVRecord record : parser) {
+          batchArgs.add(
+              new Object[] {
+                modelInfo.getModelId(),
+                epoch,
+                aAcc,
+                mFscore,
+                mPrecision,
+                mRecall,
+                mIoU,
+                mAcc,
+                changed_fscore,
+                changed_precision,
+                changed_recall,
+                unchanged_fscore,
+                unchanged_precision,
+                unchanged_recall
+              });
+        }
 
-        int epoch = Integer.parseInt(record.get("Epoch"));
-        float aAcc = Float.parseFloat(record.get("aAcc"));
-        float mFscore = Float.parseFloat(record.get("mFscore"));
-        float mPrecision = Float.parseFloat(record.get("mPrecision"));
-        float mRecall = Float.parseFloat(record.get("mRecall"));
-        float mIoU = Float.parseFloat(record.get("mIoU"));
-        float mAcc = Float.parseFloat(record.get("mAcc"));
-        float changed_fscore = Float.parseFloat(record.get("changed_fscore"));
-        float changed_precision = Float.parseFloat(record.get("changed_precision"));
-        float changed_recall = Float.parseFloat(record.get("changed_recall"));
-        float unchanged_fscore = Float.parseFloat(record.get("unchanged_fscore"));
-        float unchanged_precision = Float.parseFloat(record.get("unchanged_precision"));
-        float unchanged_recall = Float.parseFloat(record.get("unchanged_recall"));
+        modelTrainMetricsJobCoreService.insertModelMetricsValidation(batchArgs);
 
-        batchArgs.add(
-            new Object[] {
-              modelIds.getFirst(),
-              epoch,
-              aAcc,
-              mFscore,
-              mPrecision,
-              mRecall,
-              mIoU,
-              mAcc,
-              changed_fscore,
-              changed_precision,
-              changed_recall,
-              unchanged_fscore,
-              unchanged_precision,
-              unchanged_recall
-            });
+      } catch (IOException e) {
+        throw new RuntimeException(e);
       }
 
-      modelTrainMetricsJobCoreService.insertModelMetricsValidation(batchArgs);
-
-    } catch (IOException e) {
-      throw new RuntimeException(e);
+      modelTrainMetricsJobCoreService.updateModelMetricsTrainSaveYn(
+          modelInfo.getModelId(), "step1");
     }
-
-    modelTrainMetricsJobCoreService.updateModelMetricsTrainSaveYn(modelIds.getFirst(), "step1");
   }
 }

src/main/java/com/kamco/cd/training/train/service/TrainJobService.java

@@ -3,14 +3,17 @@ package com.kamco.cd.training.train.service;
 import com.fasterxml.jackson.databind.ObjectMapper;
 import com.kamco.cd.training.common.enums.TrainStatusType;
 import com.kamco.cd.training.model.dto.ModelTrainMngDto;
+import com.kamco.cd.training.model.service.TmpDatasetService;
 import com.kamco.cd.training.postgres.core.ModelTrainJobCoreService;
 import com.kamco.cd.training.postgres.core.ModelTrainMngCoreService;
 import com.kamco.cd.training.train.dto.ModelTrainJobQueuedEvent;
 import com.kamco.cd.training.train.dto.TrainRunRequest;
+import java.io.IOException;
 import java.nio.file.Files;
 import java.nio.file.Path;
 import java.nio.file.Paths;
 import java.time.ZonedDateTime;
+import java.util.List;
 import java.util.Map;
 import java.util.UUID;
 import lombok.RequiredArgsConstructor;
@@ -29,6 +32,7 @@ public class TrainJobService {
   private final DockerTrainService dockerTrainService;
   private final ObjectMapper objectMapper;
   private final ApplicationEventPublisher eventPublisher;
+  private final TmpDatasetService tmpDatasetService;
 
   // 학습 결과가 저장될 호스트 디렉토리
   @Value("${train.docker.responseDir}")
@@ -210,4 +214,26 @@ public class TrainJobService {
       return null;
     }
   }
+
+  public UUID createTmpFile(UUID modelUuid) {
+    UUID tmpUuid = UUID.randomUUID();
+    String raw = tmpUuid.toString().toUpperCase().replace("-", "");
+
+    Long modelId = modelTrainMngCoreService.findModelIdByUuid(modelUuid);
+    List<Long> datasetIds = modelTrainMngCoreService.findModelDatasetMapp(modelId);
+
+    List<String> uids = modelTrainMngCoreService.findDatasetUid(datasetIds);
+
+    try {
+      // 데이터셋 심볼링크 생성
+      Path path = tmpDatasetService.buildTmpDatasetSymlink(raw, uids);
+      ModelTrainMngDto.UpdateReq updateReq = new ModelTrainMngDto.UpdateReq();
+      updateReq.setRequestPath(path.toString());
+      modelTrainMngCoreService.updateModelMaster(modelId, updateReq);
+    } catch (IOException e) {
+      throw new RuntimeException(e);
+    }
+
+    return modelUuid;
+  }
 }

src/main/resources/application-prod.yml

@@ -4,7 +4,7 @@ spring:
       on-profile: prod
 
   jpa:
-    show-sql: false
+    show-sql: true
     hibernate:
       ddl-auto: validate
     properties:
@@ -12,41 +12,43 @@ spring:
         default_batch_fetch_size: 100     # ✅ 성능 - N+1 쿼리 방지
         order_updates: true               # ✅ 성능 - 업데이트 순서 정렬로 데드락 방지
         use_sql_comments: true            # ⚠️ 선택 - SQL에 주석 추가 (디버깅용)
+        format_sql: true                  # ⚠️ 선택 - SQL 포맷팅 (가독성)
 
   datasource:
-    url: jdbc:postgresql://10.100.0.10:25432/temp
-    username: temp
-    password: temp123!
+    url: jdbc:postgresql://127.0.01:15432/kamco_training_db
+    #   url: jdbc:postgresql://localhost:15432/kamco_training_db
+    username: kamco_training_user
+    password: kamco_training_user_2025_!@#
     hikari:
       minimum-idle: 10
       maximum-pool-size: 20
+      connection-timeout: 60000        # 60초 연결 타임아웃
+      idle-timeout: 300000            # 5분 유휴 타임아웃
+      max-lifetime: 1800000           # 30분 최대 수명
+      leak-detection-threshold: 60000  # 연결 누수 감지
+
+  transaction:
+    default-timeout: 300              # 5분 트랜잭션 타임아웃
 
 jwt:
-  secret: "kamco_token_prod_dfc6446d-68fc-4eba-a2ff-c80a14a0bf3a"
+  secret: "kamco_token_dev_dfc6446d-68fc-4eba-a2ff-c80a14a0bf3a"
   access-token-validity-in-ms: 86400000      # 1일
   refresh-token-validity-in-ms: 604800000    # 7일
 
 token:
   refresh-cookie-name: kamco   # 개발용 쿠키 이름
-  refresh-cookie-secure: true   # 로컬 http 테스트면 false
+  refresh-cookie-secure: false   # 로컬 http 테스트면 false
+
+springdoc:
+  swagger-ui:
+    persist-authorization: true # 스웨거 새로고침해도 토큰 유지, 로컬스토리지에 저장
 
 member:
   init_password: kamco1234!
 
-swagger:
-  local-port: 9080
-
-file:
-  sync-root-dir: /app/original-images/
-  sync-tmp-dir: ${file.sync-root-dir}tmp/
-  sync-file-extention: tfw,tif
-
-  dataset-dir: /home/kcomu/data/request/
-  dataset-tmp-dir: ${file.dataset-dir}tmp/
-
 train:
   docker:
-    image: "kamco-cd-train:love_latest"
+    image: "kamco-cd-train:latest"
     requestDir: "/home/kcomu/data/request"
     responseDir: "/home/kcomu/data/response"
     containerPrefix: "kamco-cd-train"

src/main/resources/db/migration/dump-kamco_training_db-202512261611.sql

@@ -9667,7 +9667,7 @@ INSERT INTO public.tb_audit_log VALUES (1813, 3, 'CREATE', 'SUCCESS', 'SYSTEM',
 INSERT INTO public.tb_audit_log VALUES (1814, 3, 'READ', 'SUCCESS', 'SYSTEM', '127.0.0.1', NULL, '2025-12-26 00:44:49.832926+00', NULL, '', '/api/models/train');
 INSERT INTO public.tb_audit_log VALUES (1815, NULL, 'CREATE', 'FAILED', 'SYSTEM', '127.0.0.1', NULL, '2025-12-26 00:53:40.899603+00', 467, '{
   "username": "1234567",
-  "password":"****" 
+  "password":"****"
 }', '/api/auth/signin');
 INSERT INTO public.tb_audit_log VALUES (1816, NULL, 'CREATE', 'SUCCESS', 'SYSTEM', '127.0.0.1', NULL, '2025-12-26 00:55:27.731595+00', NULL, '{
   "username": "1234567",
@@ -30396,6 +30396,8 @@ ALTER TABLE ONLY public.tb_menu
     ADD CONSTRAINT fksw914diut87r7lfykekc7xm2a FOREIGN KEY (parent_menu_uid) REFERENCES public.tb_menu(menu_uid);
 
 
+
+
 -- Completed on 2025-12-26 16:11:11 KST
 
 --
@@ -30404,3 +30406,5 @@ ALTER TABLE ONLY public.tb_menu
 
 \unrestrict IYrUYfSgA4Fo2gubHcb84jDXfbBZEIiOZnyLtZgnMi641GaRQa5QDogarpTr7IG
 
+
+