diff --git a/src/main/java/com/kamco/cd/training/model/service/TmpDatasetService.java b/src/main/java/com/kamco/cd/training/model/service/TmpDatasetService.java index 1edfe52..428ac23 100644 --- a/src/main/java/com/kamco/cd/training/model/service/TmpDatasetService.java +++ b/src/main/java/com/kamco/cd/training/model/service/TmpDatasetService.java @@ -16,6 +16,9 @@ public class TmpDatasetService { @Value("${train.docker.requestDir}") private String requestDir; + @Value("${train.docker.basePath}") + private String trainBaseDir; + public String buildTmpDatasetSymlink(String uid, List datasetUids) throws IOException { log.info("========== buildTmpDatasetHardlink START =========="); @@ -24,7 +27,7 @@ public class TmpDatasetService { log.info("requestDir(raw)={}", requestDir); Path BASE = toPath(requestDir); - Path tmp = Path.of("/home/kcomu/data", "tmp", uid); + Path tmp = Path.of(trainBaseDir, "tmp", uid); log.info("BASE={}", BASE); log.info("BASE exists? {}", Files.isDirectory(BASE)); @@ -33,7 +36,7 @@ public class TmpDatasetService { long noDir = 0, scannedDirs = 0, regularFiles = 0, hardlinksMade = 0; // tmp 디렉토리 준비 - for (String type : List.of("train", "val")) { + for (String type : List.of("train", "val", "test")) { for (String part : List.of("input1", "input2", "label")) { Path dir = tmp.resolve(type).resolve(part); Files.createDirectories(dir); diff --git a/src/main/resources/application-dev.yml b/src/main/resources/application-dev.yml index d418625..f678018 100644 --- a/src/main/resources/application-dev.yml +++ b/src/main/resources/application-dev.yml @@ -63,9 +63,10 @@ file: train: docker: - image: "kamco-cd-train:latest" - requestDir: "/home/kcomu/data/request" - responseDir: "/home/kcomu/data/response" - containerPrefix: "kamco-cd-train" - shmSize: "16g" + image: kamco-cd-train:latest + requestDir: /home/kcomu/data/request + responseDir: /home/kcomu/data/response + basePath: /home/kcomu/data + containerPrefix: kamco-cd-train + shmSize: 16g ipcHost: true diff --git a/src/main/resources/application-prod.yml b/src/main/resources/application-prod.yml index 9d8b40d..754c90f 100644 --- a/src/main/resources/application-prod.yml +++ b/src/main/resources/application-prod.yml @@ -62,10 +62,11 @@ file: train: docker: - image: "kamco-cd-train:latest" - requestDir: "/home/kcomu/data/request" - responseDir: "/home/kcomu/data/response" - containerPrefix: "kamco-cd-train" - shmSize: "16g" + image: kamco-cd-train:latest + requestDir: /home/kcomu/data/request + responseDir: /home/kcomu/data/response + basePath: /home/kcomu/data + containerPrefix: kamco-cd-train + shmSize: 16g ipcHost: true