하이퍼 파라미터 수정

2026-02-13 14:42:39 +09:00
parent 28919345c2
commit 5b074bdb81
1 changed files with 3 additions and 1 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,7 +310,9 @@ public class DockerTrainService {
    addArg(c, "--saturation-range", req.getSaturationRange());
    addArg(c, "--hue-delta", req.getHueDelta());

-    addArg(c, "--resume", req.getResumeFrom());
+    c.add("--resume");
+    addArg(c, "--load-from", req.getResumeFrom());
+
    addArg(c, "--save-interval", 1);
    return c;
  }