From 564a99448c3da65b8dd68867ab75f3e510cbe217 Mon Sep 17 00:00:00 2001 From: teddy Date: Fri, 20 Feb 2026 15:41:34 +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 --- .../cd/training/train/service/DockerTrainService.java | 11 +++++------ 1 file changed, 5 insertions(+), 6 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 96be3f9..2e5d7a4 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 @@ -416,9 +416,8 @@ public class DockerTrainService { if (epoch == null || epoch <= 0) throw new IllegalArgumentException("epoch must be > 0"); Path epochPath = Paths.get(responseDir, req.getOutputFolder()); - - Path checkpoint = findCheckpoint(epochPath, epoch); - String modelFile = checkpoint.toString(); + // 결과 폴더에 파라미터로 받은 베스트 epoch이 best_changed_fscore_epoch_ 로 시작하는 파일이 있는지 확인 후 pth 파일명 반환 + String modelFile = findCheckpoint(epochPath, epoch); List c = new ArrayList<>(); @@ -450,7 +449,7 @@ public class DockerTrainService { return c; } - public Path findCheckpoint(Path dir, int epoch) { + public String findCheckpoint(Path dir, int epoch) { String bestFileName = String.format("best_changed_fscore_epoch_%d.pth", epoch); String normalFileName = String.format("epoch_%d.pth", epoch); @@ -460,12 +459,12 @@ public class DockerTrainService { // 1. best 파일이 존재하면 그거 사용 if (Files.isRegularFile(bestPath)) { - return bestPath; + return bestFileName; } // 2. 없으면 일반 epoch 파일 사용 if (Files.isRegularFile(normalPath)) { - return normalPath; + return normalFileName; } throw new IllegalStateException("Checkpoint 파일이 없습니다. epoch=" + epoch); -- 2.49.1