train 결과 +1 했던 거 제거하기

2026-02-18 14:50:53 +09:00
parent d9da0d4610
commit 99a4597b5f
1 changed files with 1 additions and 1 deletions
--- 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"));