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

ai 학습실행 run command 수정

Browse Source
This commit is contained in:
Gina(박가연)
2026-03-10 22:48:10 +09:00
parent 7d5581f60c
commit f3e5347335
1 changed files with 18 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -46,6 +46,10 @@ public class DockerTrainService {
@Value("${train.docker.shmSize:16g}") @Value("${train.docker.shmSize:16g}")
private String shmSize; private String shmSize;
// data 경로 request,response 상위 폴더
@Value("${train.docker.basePath}")
private String basePath;
// IPC host 사용 여부 // IPC host 사용 여부
@Value("${train.docker.ipcHost:true}") @Value("${train.docker.ipcHost:true}")
private boolean ipcHost; private boolean ipcHost;
@@ -224,7 +228,8 @@ public class DockerTrainService {
// GPU 전체 사용 // GPU 전체 사용
c.add("--gpus"); c.add("--gpus");
c.add("all"); c.add("1"); // 학습서버 GPU 1개인 곳은 1이어야 함
// c.add("all");
// IPC host 사용 여부 // IPC host 사용 여부
if (ipcHost) { if (ipcHost) {
@@ -254,7 +259,9 @@ public class DockerTrainService {
// 요청/결과 디렉토리 볼륨 마운트 // 요청/결과 디렉토리 볼륨 마운트
c.add("-v"); c.add("-v");
c.add("/home/kcomu/data" + "/tmp:/data"); c.add(basePath + ":" + basePath); // 심볼릭 링크와 연결되는 실제 파일 경로도 마운트를 해줘야 함
c.add("-v");
c.add(basePath + "/tmp:/data");
c.add("-v"); c.add("-v");
c.add(responseDir + ":/checkpoints"); c.add(responseDir + ":/checkpoints");
@@ -273,8 +280,10 @@ public class DockerTrainService {
addArg(c, "--output-folder", req.getOutputFolder()); addArg(c, "--output-folder", req.getOutputFolder());
addArg(c, "--input-size", req.getInputSize()); addArg(c, "--input-size", req.getInputSize());
addArg(c, "--crop-size", req.getCropSize()); addArg(c, "--crop-size", req.getCropSize());
addArg(c, "--batch-size", req.getBatchSize()); // addArg(c, "--batch-size", req.getBatchSize());
addArg(c, "--gpu-ids", req.getGpuIds()); // null // addArg(c, "--gpu-ids", req.getGpuIds()); // null
addArg(c, "--batch-size", 2); // 학습서버 GPU 1개인 곳은 batch-size:2 까지만 가능
addArg(c, "--gpu-ids", "0"); // 학습서버 GPU 1개인 곳은 0이어야 함
addArg(c, "--lr", req.getLearningRate()); addArg(c, "--lr", req.getLearningRate());
addArg(c, "--backbone", req.getBackbone()); addArg(c, "--backbone", req.getBackbone());
addArg(c, "--epochs", req.getEpochs()); addArg(c, "--epochs", req.getEpochs());
@@ -439,12 +448,15 @@ public class DockerTrainService {
c.add("run"); c.add("run");
c.add("--rm"); c.add("--rm");
c.add("--gpus"); c.add("--gpus");
c.add("all"); c.add("1"); // 학습서버 GPU 1개인 곳은 1이어야 함
// c.add("all");
c.add("--ipc=host"); c.add("--ipc=host");
c.add("--shm-size=" + shmSize); c.add("--shm-size=" + shmSize);
c.add("-v"); c.add("-v");
c.add("/home/kcomu/data" + "/tmp:/data"); c.add(basePath + ":" + basePath); // 심볼릭 링크와 연결되는 실제 파일 경로도 마운트를 해줘야 함
c.add("-v");
c.add(basePath + "/tmp:/data");
c.add("-v"); c.add("-v");
c.add(responseDir + ":/checkpoints"); c.add(responseDir + ":/checkpoints");