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 b27e5a2..be9486b 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 @@ -169,7 +169,7 @@ public class DockerTrainService { // 컨테이너 이름 지정 c.add("--name"); - c.add(containerName + "-" + req.getOutputFolder().substring(0, 8)); + c.add(containerName + "-" + req.getUuid().substring(0, 8)); // 실행 종료 시 자동 삭제 c.add("--rm"); @@ -221,7 +221,7 @@ public class DockerTrainService { c.add("/workspace/change-detection-code/train_wrapper.py"); // ===== 기본 파라미터 ===== - addArg(c, "--dataset-folder", req.getDatasetFolder()); + addArg(c, "--dataset-folder", "4BDBBDF99D04477A927CC9EBA760B845" /*req.getDatasetFolder()*/); addArg(c, "--output-folder", req.getOutputFolder()); addArg(c, "--input-size", req.getInputSize()); addArg(c, "--crop-size", req.getCropSize());