From 38ae6e55757e0415c228bbe88c883891adaa1a2a Mon Sep 17 00:00:00 2001 From: teddy Date: Fri, 20 Feb 2026 15:31:33 +0900 Subject: [PATCH] =?UTF-8?q?best=20epoch=20=ED=8C=8C=EC=9D=BC=20=EC=84=A0?= =?UTF-8?q?=ED=83=9D=20=EC=88=98=EC=A0=95?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../training/train/service/DockerTrainService.java | 12 +++++------- 1 file changed, 5 insertions(+), 7 deletions(-) diff --git a/src/main/java/com/kamco/cd/training/train/service/DockerTrainService.java b/src/main/java/com/kamco/cd/training/train/service/DockerTrainService.java index b1c029b..96be3f9 100644 --- a/src/main/java/com/kamco/cd/training/train/service/DockerTrainService.java +++ b/src/main/java/com/kamco/cd/training/train/service/DockerTrainService.java @@ -9,6 +9,7 @@ import java.io.InputStreamReader; import java.nio.charset.StandardCharsets; import java.nio.file.Files; import java.nio.file.Path; +import java.nio.file.Paths; import java.util.ArrayList; import java.util.List; import java.util.concurrent.TimeUnit; @@ -414,12 +415,10 @@ public class DockerTrainService { if (uuid == null || uuid.isBlank()) throw new IllegalArgumentException("uuid is required"); if (epoch == null || epoch <= 0) throw new IllegalArgumentException("epoch must be > 0"); - String modelFile = - String.valueOf(findCheckpoint(Path.of(responseDir + req.getOutputFolder()), epoch)); + Path epochPath = Paths.get(responseDir, req.getOutputFolder()); - if (modelFile == null || modelFile.isBlank()) { - throw new IllegalArgumentException("best model file is required"); - } + Path checkpoint = findCheckpoint(epochPath, epoch); + String modelFile = checkpoint.toString(); List c = new ArrayList<>(); @@ -469,7 +468,6 @@ public class DockerTrainService { return normalPath; } - // 둘 다 없으면 null 또는 예외 - return null; + throw new IllegalStateException("Checkpoint 파일이 없습니다. epoch=" + epoch); } }