Files
kamco-cd-cron/kamco-make-dataset-generation/src/main/resources/application-prod.yml
2026-02-09 20:29:50 +09:00

32 lines
728 B
YAML

spring:
datasource:
url: jdbc:postgresql://127.0.0.1:15432/kamco_cds
username: kamco_cds
password: kamco_cds_Q!W@E#R$
hikari:
minimum-idle: 2
maximum-pool-size: 5
batch:
job:
name: exportGeoJsonJob # 기본 실행 Job 지정
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