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

batch_size

Browse Source
This commit is contained in:
teddy
2026-04-06 21:41:11 +09:00
parent 38906a5795
commit a0f4d0b8e2
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -62,7 +62,7 @@ public class DockerTrainService {
@Value("${hyper.parameter.gpu-ids}") @Value("${hyper.parameter.gpu-ids}")
private String hyperGpuIds; private String hyperGpuIds;
@Value("${hyper.parameter.batch_size}") @Value("${hyper.parameter.batch-size}")
private Integer batchSize; private Integer batchSize;
private final ModelTrainJobCoreService modelTrainJobCoreService; private final ModelTrainJobCoreService modelTrainJobCoreService;