MVPTeam/kamco-train-api
6
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Compare commits

base: MVPTeam:96cb7d2f2347a6067820cf2fbfdba206697fe2be
Branches Tags
MVPTeam:develop MVPTeam:feat/training_260324 MVPTeam:feat/training_260303 MVPTeam:feat/training_260202 MVPTeam:feat/dean/hyperparam_with_modelType
..
compare: MVPTeam:f4d491ed94d4fd70f2f00ccd34322e57c307c305
Branches Tags
MVPTeam:develop MVPTeam:feat/training_260324 MVPTeam:feat/training_260303 MVPTeam:feat/training_260202 MVPTeam:feat/dean/hyperparam_with_modelType

2 Commits
96cb7d2f23 ... f4d491ed94

Author SHA1 Message Date
teddy
f4d491ed94 Merge pull request '하이퍼 파라미터 수정' (#105) from feat/training_260202 into develop 
Reviewed-on: #105
 2026-02-13 14:31:13 +09:00
teddy
aa0552aaa7 하이퍼 파라미터 수정 2026-02-13 14:30:45 +09:00
1 changed files with 3 additions and 3 deletions
Expand all files Collapse all files

6
src/main/java/com/kamco/cd/training/train/service/DockerTrainService.java
View File

@@ -302,12 +302,12 @@ public class DockerTrainService {
// ===== Augmentation =====
addArg(c, "--rot-prob", req.getRotProb());
// addArg(c, "--rot-degree", req.getRotDegree()); // TODO AI 수정되면 주석 해제
addArg(c, "--rot-degree", req.getRotDegree());
addArg(c, "--flip-prob", req.getFlipProb());
addArg(c, "--exchange-prob", req.getExchangeProb());
addArg(c, "--brightness-delta", req.getBrightnessDelta());
// addArg(c, "--contrast-range", req.getContrastRange()); // TODO AI 수정되면 주석 해제
// addArg(c, "--saturation-range", req.getSaturationRange()); // TODO AI 수정되면 주석 해제
addArg(c, "--contrast-range", req.getContrastRange());
addArg(c, "--saturation-range", req.getSaturationRange());
addArg(c, "--hue-delta", req.getHueDelta());
addArg(c, "--resume", req.getResumeFrom());