Compare commits
3 Commits
feat/rebui
...
d99e18b38c
| Author | SHA1 | Date | |
|---|---|---|---|
| d99e18b38c | |||
| d6aa612494 | |||
| 8def356323 |
@@ -51,10 +51,10 @@ public class ModelTrainMngService {
|
|||||||
|
|
||||||
@Value("${train.docker.response_dir}")
|
@Value("${train.docker.response_dir}")
|
||||||
private String responseDir;
|
private String responseDir;
|
||||||
|
|
||||||
@Value("${train.docker.symbolic_link_dir}")
|
@Value("${train.docker.symbolic_link_dir}")
|
||||||
private String symbolicDir;
|
private String symbolicDir;
|
||||||
|
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* 모델학습 조회
|
* 모델학습 조회
|
||||||
*
|
*
|
||||||
|
|||||||
@@ -262,9 +262,9 @@ public class DockerTrainService {
|
|||||||
c.add("-v");
|
c.add("-v");
|
||||||
c.add(basePath + ":" + basePath); // 심볼릭 링크와 연결되는 실제 파일 경로도 마운트를 해줘야 함
|
c.add(basePath + ":" + basePath); // 심볼릭 링크와 연결되는 실제 파일 경로도 마운트를 해줘야 함
|
||||||
c.add("-v");
|
c.add("-v");
|
||||||
c.add(symbolicDir + ":/data"); //요청할경로
|
c.add(symbolicDir + ":/data"); // 요청할경로
|
||||||
c.add("-v");
|
c.add("-v");
|
||||||
c.add(responseDir + ":/checkpoints"); //저장될경로
|
c.add(responseDir + ":/checkpoints"); // 저장될경로
|
||||||
|
|
||||||
// 표준입력 유지 (-it 대신 -i만 사용)
|
// 표준입력 유지 (-it 대신 -i만 사용)
|
||||||
c.add("-i");
|
c.add("-i");
|
||||||
|
|||||||
@@ -85,34 +85,20 @@ public class ModelTrainMetricsJobService {
|
|||||||
|
|
||||||
int epoch = Integer.parseInt(record.get("Epoch"));
|
int epoch = Integer.parseInt(record.get("Epoch"));
|
||||||
|
|
||||||
float aAcc = parseFloatSafe(record.get("aAcc"));
|
Float aAcc = parseFloatSafe(record.get("aAcc"));
|
||||||
float mFscore = parseFloatSafe(record.get("mFscore"));
|
Float mFscore = parseFloatSafe(record.get("mFscore"));
|
||||||
float mPrecision = parseFloatSafe(record.get("mPrecision"));
|
Float mPrecision = parseFloatSafe(record.get("mPrecision"));
|
||||||
float mRecall = parseFloatSafe(record.get("mRecall"));
|
Float mRecall = parseFloatSafe(record.get("mRecall"));
|
||||||
float mIoU = parseFloatSafe(record.get("mIoU"));
|
Float mIoU = parseFloatSafe(record.get("mIoU"));
|
||||||
float mAcc = parseFloatSafe(record.get("mAcc"));
|
Float mAcc = parseFloatSafe(record.get("mAcc"));
|
||||||
|
|
||||||
float changed_fscore = parseFloatSafe(record.get("changed_fscore"));
|
Float changed_fscore = parseFloatSafe(record.get("changed_fscore"));
|
||||||
float changed_precision = parseFloatSafe(record.get("changed_precision"));
|
Float changed_precision = parseFloatSafe(record.get("changed_precision"));
|
||||||
float changed_recall = parseFloatSafe(record.get("changed_recall"));
|
Float changed_recall = parseFloatSafe(record.get("changed_recall"));
|
||||||
|
|
||||||
float unchanged_fscore = parseFloatSafe(record.get("unchanged_fscore"));
|
Float unchanged_fscore = parseFloatSafe(record.get("unchanged_fscore"));
|
||||||
float unchanged_precision = parseFloatSafe(record.get("unchanged_precision"));
|
Float unchanged_precision = parseFloatSafe(record.get("unchanged_precision"));
|
||||||
float unchanged_recall = parseFloatSafe(record.get("unchanged_recall"));
|
Float unchanged_recall = parseFloatSafe(record.get("unchanged_recall"));
|
||||||
// int epoch = Integer.parseInt(record.get("Epoch"));
|
|
||||||
// float aAcc = Float.parseFloat(record.get("aAcc"));
|
|
||||||
// float mFscore = Float.parseFloat(record.get("mFscore"));
|
|
||||||
// float mPrecision = Float.parseFloat(record.get("mPrecision"));
|
|
||||||
// float mRecall = Float.parseFloat(record.get("mRecall"));
|
|
||||||
// float mIoU = Float.parseFloat(record.get("mIoU"));
|
|
||||||
// float mAcc = Float.parseFloat(record.get("mAcc"));
|
|
||||||
// float changed_fscore = Float.parseFloat(record.get("changed_fscore"));
|
|
||||||
// float changed_precision = Float.parseFloat(record.get("changed_precision"));
|
|
||||||
// float changed_recall = Float.parseFloat(record.get("changed_recall"));
|
|
||||||
// float unchanged_fscore = Float.parseFloat(record.get("unchanged_fscore"));
|
|
||||||
// float unchanged_precision =
|
|
||||||
// Float.parseFloat(record.get("unchanged_precision"));
|
|
||||||
// float unchanged_recall = Float.parseFloat(record.get("unchanged_recall"));
|
|
||||||
|
|
||||||
batchArgs.add(
|
batchArgs.add(
|
||||||
new Object[] {
|
new Object[] {
|
||||||
|
|||||||
@@ -19,6 +19,7 @@ public class TmpDatasetService {
|
|||||||
|
|
||||||
@Value("${train.docker.symbolic_link_dir}")
|
@Value("${train.docker.symbolic_link_dir}")
|
||||||
private String symbolicDir;
|
private String symbolicDir;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* train, val, test 폴더별로 link
|
* train, val, test 폴더별로 link
|
||||||
*
|
*
|
||||||
|
|||||||
Reference in New Issue
Block a user