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