Split the function

2026-02-09 20:26:38 +09:00
parent be4840bb58
commit 703c25aadf
4 changed files with 19 additions and 153 deletions
--- a/kamco-make-dataset-generation/src/main/resources/application-prod.yml
+++ b/kamco-make-dataset-generation/src/main/resources/application-prod.yml
@@ -9,3 +9,20 @@ spring:

 training-data:
  geojson-dir: /kamco-nfs/dataset
+
+# Train Model Docker Configuration
+train-data:
+  docker:
+    image: kamco-cd-train:latest
+    data-volume: /kamco-nfs/dataset:/data
+    checkpoints-volume: /kamco-nfs/checkpoints:/checkpoints
+    dataset-folder: /data/dataset
+    output-folder: /data/output
+    input-size: "512"
+    crop-size: "256"
+    batch-size: 8
+    gpu-ids: "0,1,2,3"
+    gpus: 4
+    lr: "0.001"
+    backbone: resnet50
+    epochs: 100