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 428ac23..e0d96ed 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 @@ -37,7 +37,7 @@ public class TmpDatasetService { // tmp 디렉토리 준비 for (String type : List.of("train", "val", "test")) { - for (String part : List.of("input1", "input2", "label")) { + for (String part : List.of("input1", "input2", "label", "label-json")) { Path dir = tmp.resolve(type).resolve(part); Files.createDirectories(dir); log.info("createDirectories: {}", dir); @@ -70,7 +70,7 @@ public class TmpDatasetService { log.info("---- dataset id={} srcRoot={} exists? {}", id, srcRoot, Files.isDirectory(srcRoot)); for (String type : List.of("train", "val", "test")) { - for (String part : List.of("input1", "input2", "label")) { + for (String part : List.of("input1", "input2", "label", "label-json")) { Path srcDir = srcRoot.resolve(type).resolve(part);