diff --git a/src/main/java/com/kamco/cd/training/train/service/TrainJobService.java b/src/main/java/com/kamco/cd/training/train/service/TrainJobService.java
index 270a029..69ba7fb 100644
--- a/src/main/java/com/kamco/cd/training/train/service/TrainJobService.java
+++ b/src/main/java/com/kamco/cd/training/train/service/TrainJobService.java
@@ -207,7 +207,9 @@ public class TrainJobService {
 
     try {
       String ckptFile = Files.readString(last).trim(); // epoch_10.pth
+      ckptFile = ckptFile.replace("/checkpoints", responseDir);
       log.info("resume ckptFile: {}", ckptFile);
+
       if (ckptFile.isEmpty()) return null;
 
       Path ckptHost = outDir.resolve(ckptFile);