하이퍼 파라미터 수정

2026-02-13 14:46:59 +09:00
parent 5b074bdb81
commit 0798b352c7
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;