From e6caea05b3e1a667450601ad36bec54a2099e997 Mon Sep 17 00:00:00 2001
From: teddy <teddy.kim.kim@tf.dabeeo.com>
Date: Mon, 23 Feb 2026 15:19:40 +0900
Subject: [PATCH] =?UTF-8?q?=ED=95=98=EC=9D=B4=ED=8D=BC=20=ED=8C=8C?=
 =?UTF-8?q?=EB=9D=BC=EB=AF=B8=ED=84=B0=20=EC=82=AC=EC=9A=A9=ED=9A=8C?=
 =?UTF-8?q?=EC=88=98=20=EC=B9=B4=EC=9A=B4=ED=8A=B8=20=EA=B8=B0=EB=8A=A5=20?=
 =?UTF-8?q?=EC=B6=94=EA=B0=80=20=EB=B0=8F=20=EC=A1=B0=ED=9A=8C=20=EC=88=98?=
 =?UTF-8?q?=EC=A0=95?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../training/postgres/core/HyperParamCoreService.java  |  4 ++--
 .../postgres/core/ModelTrainMngCoreService.java        |  3 +++
 .../postgres/entity/ModelHyperParamEntity.java         |  3 +++
 .../cd/training/train/service/TrainJobService.java     | 10 ++--------
 4 files changed, 10 insertions(+), 10 deletions(-)

diff --git a/src/main/java/com/kamco/cd/training/postgres/core/HyperParamCoreService.java b/src/main/java/com/kamco/cd/training/postgres/core/HyperParamCoreService.java
index cffd16e..5098c4b 100644
--- a/src/main/java/com/kamco/cd/training/postgres/core/HyperParamCoreService.java
+++ b/src/main/java/com/kamco/cd/training/postgres/core/HyperParamCoreService.java
@@ -34,7 +34,7 @@ public class HyperParamCoreService {
 
     ModelHyperParamEntity entity = new ModelHyperParamEntity();
     entity.setHyperVer(firstVersion);
-
+    entity.setTotalUseCnt(entity.getTotalUseCnt() == null ? 1 : entity.getTotalUseCnt() + 1);
     applyHyperParam(entity, createReq);
 
     // user
@@ -172,7 +172,7 @@ public class HyperParamCoreService {
     } else {
       entity.setCropSize("256,256");
     }
-    //    entity.setCropSize(src.getCropSize());
+    entity.setCropSize(src.getCropSize());
 
     // Important
     entity.setModelType(model); // 20250212 modeltype추가
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 bb56dbd..1563b7c 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
@@ -104,6 +104,9 @@ public class ModelTrainMngCoreService {
     if (hyperParamEntity == null || hyperParamEntity.getHyperVer() == null) {
       throw new CustomApiException("NOT_FOUND_DATA", HttpStatus.NOT_FOUND);
     }
+    // 하이퍼 파라미터 사용회수 업데이트
+    hyperParamEntity.setTotalUseCnt(
+        hyperParamEntity.getTotalUseCnt() == null ? 1 : hyperParamEntity.getTotalUseCnt() + 1);
 
     String modelVer =
         String.join(
diff --git a/src/main/java/com/kamco/cd/training/postgres/entity/ModelHyperParamEntity.java b/src/main/java/com/kamco/cd/training/postgres/entity/ModelHyperParamEntity.java
index d0cd472..fb250b1 100644
--- a/src/main/java/com/kamco/cd/training/postgres/entity/ModelHyperParamEntity.java
+++ b/src/main/java/com/kamco/cd/training/postgres/entity/ModelHyperParamEntity.java
@@ -310,6 +310,9 @@ public class ModelHyperParamEntity {
   @Column(name = "default_param")
   private Boolean isDefault = false;
 
+  @Column(name = "total_use_cnt")
+  private Integer totalUseCnt = 0;
+
   public HyperParamDto.Basic toDto() {
     return new HyperParamDto.Basic(
         this.modelType,
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 68214a9..b5dd84e 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
@@ -296,14 +296,8 @@ public class TrainJobService {
           e);
 
       // 런타임 예외로 래핑하되, 메시지에 핵심 정보 포함
-      throw new IllegalStateException(
-          "tmp dataset build failed: modelUuid="
-              + modelUuid
-              + ", modelId="
-              + modelId
-              + ", tmpRaw="
-              + raw,
-          e);
+      throw new CustomApiException(
+          "INTERNAL_SERVER_ERROR", HttpStatus.INTERNAL_SERVER_ERROR, "임시 데이터셋 생성에 실패했습니다.");
     }
     return modelUuid;
   }