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

이어하기 수정

Browse Source
This commit is contained in:
teddy
2026-02-13 13:57:01 +09:00
parent a070566048
commit 4da477706f
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/main/java/com/kamco/cd/training/train/service/TrainJobService.java
View File

@@ -207,13 +207,15 @@ public class TrainJobService {
try { try {
String ckptFile = Files.readString(last).trim(); // epoch_10.pth String ckptFile = Files.readString(last).trim(); // epoch_10.pth
log.info("resume ckptFile: {}", ckptFile);
if (ckptFile.isEmpty()) return null; if (ckptFile.isEmpty()) return null;
Path ckptHost = outDir.resolve(ckptFile); Path ckptHost = outDir.resolve(ckptFile);
log.info("resume ckptHost: {}", ckptHost);
if (!Files.isRegularFile(ckptHost)) return null; if (!Files.isRegularFile(ckptHost)) return null;
// 컨테이너 경로 반환 // 컨테이너 경로 반환
return "/checkpoints/" + outputFolder + "/" + ckptFile; return responseDir + outputFolder + "/" + ckptFile;
} catch (Exception e) { } catch (Exception e) {
return null; return null;