batch_size

2026-04-06 21:41:11 +09:00
parent 38906a5795
commit a0f4d0b8e2
1 changed files with 1 additions and 1 deletions
--- a/src/main/java/com/kamco/cd/training/train/service/DockerTrainService.java
+++ b/src/main/java/com/kamco/cd/training/train/service/DockerTrainService.java
@@ -62,7 +62,7 @@ public class DockerTrainService {
  @Value("${hyper.parameter.gpu-ids}")
  private String hyperGpuIds;

-  @Value("${hyper.parameter.batch_size}")
+  @Value("${hyper.parameter.batch-size}")
  private Integer batchSize;

  private final ModelTrainJobCoreService modelTrainJobCoreService;