diff --git a/src/main/java/com/kamco/cd/training/model/service/ModelTrainMngService.java b/src/main/java/com/kamco/cd/training/model/service/ModelTrainMngService.java index b87b034..d4016e0 100644 --- a/src/main/java/com/kamco/cd/training/model/service/ModelTrainMngService.java +++ b/src/main/java/com/kamco/cd/training/model/service/ModelTrainMngService.java @@ -51,10 +51,10 @@ public class ModelTrainMngService { @Value("${train.docker.response_dir}") private String responseDir; + @Value("${train.docker.symbolic_link_dir}") private String symbolicDir; - /** * 모델학습 조회 * 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 7cd8ebc..1f37115 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 @@ -262,9 +262,9 @@ public class DockerTrainService { c.add("-v"); c.add(basePath + ":" + basePath); // 심볼릭 링크와 연결되는 실제 파일 경로도 마운트를 해줘야 함 c.add("-v"); - c.add(symbolicDir + ":/data"); //요청할경로 + c.add(symbolicDir + ":/data"); // 요청할경로 c.add("-v"); - c.add(responseDir + ":/checkpoints"); //저장될경로 + c.add(responseDir + ":/checkpoints"); // 저장될경로 // 표준입력 유지 (-it 대신 -i만 사용) c.add("-i"); diff --git a/src/main/java/com/kamco/cd/training/train/service/TmpDatasetService.java b/src/main/java/com/kamco/cd/training/train/service/TmpDatasetService.java index 505325d..e268177 100644 --- a/src/main/java/com/kamco/cd/training/train/service/TmpDatasetService.java +++ b/src/main/java/com/kamco/cd/training/train/service/TmpDatasetService.java @@ -19,6 +19,7 @@ public class TmpDatasetService { @Value("${train.docker.symbolic_link_dir}") private String symbolicDir; + /** * train, val, test 폴더별로 link *