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<String> 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);

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

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