MVPTeam/kamco-train-api
6
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge pull request 'feat/training_260202' (#50) from feat/training_260202 into develop

Browse Source 
Reviewed-on: #50
This commit was merged in pull request #50.
This commit is contained in:
teddy
2026-02-12 15:52:09 +09:00
parent cd0cf5726d 8fa722011c
commit 26e8e1492f
3 changed files with 13 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
src/main/java/com/kamco/cd/training/common/utils/FIleChecker.java
View File

@@ -506,11 +506,15 @@ public class FIleChecker {
try { try {
File dir = new File(targetPath); File dir = new File(targetPath);
log.info("targetPath={}", targetPath);
log.info("absolute targetPath={}", dir.getAbsolutePath());
if (!dir.exists()) { if (!dir.exists()) {
dir.mkdirs(); dir.mkdirs();
} }
File dest = new File(dir, String.valueOf(chunkIndex)); File dest = new File(dir, String.valueOf(chunkIndex));
log.info("real save path = {}", dest.getAbsolutePath());
log.info("chunkIndex={}, uploadSize={}", chunkIndex, mfile.getSize()); log.info("chunkIndex={}, uploadSize={}", chunkIndex, mfile.getSize());
log.info("savedSize={}", dest.length()); log.info("savedSize={}", dest.length());
@@ -522,6 +526,9 @@ public class FIleChecker {
log.info("after delete={}", dest.length()); log.info("after delete={}", dest.length());
mfile.transferTo(dest); mfile.transferTo(dest);
log.info("after transfer size={}", dest.length());
log.info("after transfer exists={}", dest.exists());
return true; return true;
} catch (IOException e) { } catch (IOException e) {
log.error("chunk save error", e); log.error("chunk save error", e);

2
src/main/java/com/kamco/cd/training/model/service/ModelTrainMngService.java
View File

@@ -12,7 +12,6 @@ import com.kamco.cd.training.model.dto.ModelTrainMngDto;
import com.kamco.cd.training.model.dto.ModelTrainMngDto.SearchReq; import com.kamco.cd.training.model.dto.ModelTrainMngDto.SearchReq;
import com.kamco.cd.training.postgres.core.HyperParamCoreService; import com.kamco.cd.training.postgres.core.HyperParamCoreService;
import com.kamco.cd.training.postgres.core.ModelTrainMngCoreService; import com.kamco.cd.training.postgres.core.ModelTrainMngCoreService;
import com.kamco.cd.training.train.service.TrainJobService;
import java.io.IOException; import java.io.IOException;
import java.nio.file.Path; import java.nio.file.Path;
import java.util.List; import java.util.List;
@@ -30,7 +29,6 @@ import org.springframework.transaction.annotation.Transactional;
@Slf4j @Slf4j
public class ModelTrainMngService { public class ModelTrainMngService {
private final TrainJobService trainJobService;
private final ModelTrainMngCoreService modelTrainMngCoreService; private final ModelTrainMngCoreService modelTrainMngCoreService;
private final HyperParamCoreService hyperParamCoreService; private final HyperParamCoreService hyperParamCoreService;
private final TmpDatasetService tmpDatasetService; private final TmpDatasetService tmpDatasetService;

12
src/main/java/com/kamco/cd/training/model/service/TmpDatasetService.java
View File

@@ -14,14 +14,14 @@ import org.springframework.transaction.annotation.Transactional;
@RequiredArgsConstructor @RequiredArgsConstructor
public class TmpDatasetService { public class TmpDatasetService {
@Value("${train.requestDir}") @Value("${train.docker.requestDir}")
private String requestDir; private String requestDir;
// 환경에 맞게 yml로 빼는 걸 추천
private final Path BASE = Paths.get(requestDir);
@Transactional(readOnly = true) @Transactional(readOnly = true)
public Path buildTmpDatasetSymlink(String uid, List<String> uids) throws IOException { public Path buildTmpDatasetSymlink(String uid, List<String> datasetUids) throws IOException {
// 환경에 맞게 yml로 빼는 걸 추천
Path BASE = Paths.get(requestDir);
Path tmp = BASE.resolve("tmp").resolve(uid); Path tmp = BASE.resolve("tmp").resolve(uid);
// mkdir -p "$TMP"/train/{input1,input2,label} ... // mkdir -p "$TMP"/train/{input1,input2,label} ...
@@ -31,7 +31,7 @@ public class TmpDatasetService {
} }
} }
for (String id : uids) { for (String id : datasetUids) {
Path srcRoot = BASE.resolve(id); Path srcRoot = BASE.resolve(id);
for (String type : List.of("train", "val")) { for (String type : List.of("train", "val")) {