diff --git a/src/main/java/com/kamco/cd/training/train/service/DockerTrainService.java b/src/main/java/com/kamco/cd/training/train/service/DockerTrainService.java
index 55398ea..4c3dea4 100644
--- a/src/main/java/com/kamco/cd/training/train/service/DockerTrainService.java
+++ b/src/main/java/com/kamco/cd/training/train/service/DockerTrainService.java
@@ -87,6 +87,7 @@ public class DockerTrainService {
                     log.info("[EPOCH] container={} {}/{}", containerName, currentEpoch, totalEpoch);
 
                     // TODO 실행중인 에폭 저장 필요하면 만들어야함
+                    // TODO 완료여부를 여기다가?
                     //                modelTrainMngCoreService.updateCurrentEpoch(modelId,
                     // currentEpoch, totalEpoch);
                   }