feat/training_260202 #111
@@ -70,7 +70,7 @@ public class ModelTrainMetricsJobService {
|
|||||||
|
|
||||||
for (CSVRecord record : parser) {
|
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"));
|
long iteration = Long.parseLong(record.get("Iteration"));
|
||||||
double Loss = Double.parseDouble(record.get("Loss"));
|
double Loss = Double.parseDouble(record.get("Loss"));
|
||||||
double LR = Double.parseDouble(record.get("LR"));
|
double LR = Double.parseDouble(record.get("LR"));
|
||||||
|
|||||||
Reference in New Issue
Block a user