From a0f4d0b8e2749132f1813d9cd585fefbf6cb7b8b Mon Sep 17 00:00:00 2001 From: teddy Date: Mon, 6 Apr 2026 21:41:11 +0900 Subject: [PATCH] batch_size --- .../com/kamco/cd/training/train/service/DockerTrainService.java | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/main/java/com/kamco/cd/training/train/service/DockerTrainService.java b/src/main/java/com/kamco/cd/training/train/service/DockerTrainService.java index 60ee90d..e33e83d 100644 --- 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;