From f5f8939a5c97c97ffc04282296f669bfb5cf25bb Mon Sep 17 00:00:00 2001 From: songhyeonsu Date: Fri, 8 May 2026 13:36:00 +0900 Subject: [PATCH] Wire up wandb logging for Step1 training - Global.use_wandb: True + top-level wandb.project=kr_lp_pgnet - Add wandb to setup_server.sh pip install list User must run `docker exec -it kr_lp_pgnet wandb login` once before training so the API key lands in /root/.netrc inside the container. Co-Authored-By: Claude Opus 4.7 --- configs/kr_lp_pgnet.yml | 5 +++++ scripts/setup_server.sh | 4 ++-- 2 files changed, 7 insertions(+), 2 deletions(-) diff --git a/configs/kr_lp_pgnet.yml b/configs/kr_lp_pgnet.yml index 77577be..54d7202 100644 --- a/configs/kr_lp_pgnet.yml +++ b/configs/kr_lp_pgnet.yml @@ -11,6 +11,7 @@ Global: checkpoints: save_inference_dir: use_visualdl: False + use_wandb: True infer_img: infer_visual_type: CN valid_set: partvgg @@ -21,6 +22,10 @@ Global: max_text_nums: 5 tcl_len: 64 +wandb: + project: kr_lp_pgnet + name: step1-pretrain + Architecture: model_type: e2e algorithm: PGNet diff --git a/scripts/setup_server.sh b/scripts/setup_server.sh index 315ebb0..5c00423 100755 --- a/scripts/setup_server.sh +++ b/scripts/setup_server.sh @@ -54,8 +54,8 @@ while IFS= read -r line; do $PIP install $PIP_OPTS --ignore-installed "$line" 2>/dev/null || echo " skip: $line" done < /tmp/kr_lp_req.txt -echo "[5/7] OpenCV (numpy2 호환) + numpy<2 (PaddleOCR release/2.7 호환성)" -$PIP install $PIP_OPTS 'opencv-python>=4.10' 'opencv-contrib-python>=4.10' +echo "[5/7] OpenCV (numpy2 호환) + numpy<2 (PaddleOCR release/2.7 호환성) + wandb" +$PIP install $PIP_OPTS 'opencv-python>=4.10' 'opencv-contrib-python>=4.10' wandb # imgaug 등이 numpy 1.x API(np.sctypes)에 의존하므로 numpy 1.x로 핀. # paddle 3.3.0.dev는 numpy 1.x도 호환. $PIP install $PIP_OPTS 'numpy<2' --force-reinstall