diff --git a/src/main/java/com/kamco/cd/training/train/service/ModelTrainMetricsJobService.java b/src/main/java/com/kamco/cd/training/train/service/ModelTrainMetricsJobService.java
index 4487973..933b9f1 100644
--- a/src/main/java/com/kamco/cd/training/train/service/ModelTrainMetricsJobService.java
+++ b/src/main/java/com/kamco/cd/training/train/service/ModelTrainMetricsJobService.java
@@ -70,7 +70,7 @@ public class ModelTrainMetricsJobService {
 
         for (CSVRecord record : parser) {
 
-          int epoch = Integer.parseInt(record.get("Epoch")) + 1; // TODO : 나중에 AI 개발 완료되면 -1 하기
+          int epoch = Integer.parseInt(record.get("Epoch"));
           long iteration = Long.parseLong(record.get("Iteration"));
           double Loss = Double.parseDouble(record.get("Loss"));
           double LR = Double.parseDouble(record.get("LR"));