Merge pull request '하이퍼 파라미터 수정' (#108) from feat/training_260202 into develop

Reviewed-on: #108
2026-02-13 14:47:19 +09:00
parent df09935789 0798b352c7
commit 31ac4209c3
1 changed files with 4 additions and 2 deletions
--- a/src/main/java/com/kamco/cd/training/train/service/DockerTrainService.java
+++ b/src/main/java/com/kamco/cd/training/train/service/DockerTrainService.java
@@ -310,8 +310,10 @@ public class DockerTrainService {
    addArg(c, "--saturation-range", req.getSaturationRange());
    addArg(c, "--hue-delta", req.getHueDelta());

-    c.add("--resume");
-    addArg(c, "--load-from", req.getResumeFrom());
+    if (req.getResumeFrom() != null && !req.getResumeFrom().isBlank()) {
+      c.add("--resume");
+      addArg(c, "--load-from", req.getResumeFrom());
+    }

    addArg(c, "--save-interval", 1);
    return c;