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 <noreply@anthropic.com>
main
songhyeonsu 1 month ago
parent 82c046522e
commit f5f8939a5c

@ -11,6 +11,7 @@ Global:
checkpoints: checkpoints:
save_inference_dir: save_inference_dir:
use_visualdl: False use_visualdl: False
use_wandb: True
infer_img: infer_img:
infer_visual_type: CN infer_visual_type: CN
valid_set: partvgg valid_set: partvgg
@ -21,6 +22,10 @@ Global:
max_text_nums: 5 max_text_nums: 5
tcl_len: 64 tcl_len: 64
wandb:
project: kr_lp_pgnet
name: step1-pretrain
Architecture: Architecture:
model_type: e2e model_type: e2e
algorithm: PGNet algorithm: PGNet

@ -54,8 +54,8 @@ while IFS= read -r line; do
$PIP install $PIP_OPTS --ignore-installed "$line" 2>/dev/null || echo " skip: $line" $PIP install $PIP_OPTS --ignore-installed "$line" 2>/dev/null || echo " skip: $line"
done < /tmp/kr_lp_req.txt done < /tmp/kr_lp_req.txt
echo "[5/7] OpenCV (numpy2 호환) + numpy<2 (PaddleOCR release/2.7 호환성)" 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' $PIP install $PIP_OPTS 'opencv-python>=4.10' 'opencv-contrib-python>=4.10' wandb
# imgaug 등이 numpy 1.x API(np.sctypes)에 의존하므로 numpy 1.x로 핀. # imgaug 등이 numpy 1.x API(np.sctypes)에 의존하므로 numpy 1.x로 핀.
# paddle 3.3.0.dev는 numpy 1.x도 호환. # paddle 3.3.0.dev는 numpy 1.x도 호환.
$PIP install $PIP_OPTS 'numpy<2' --force-reinstall $PIP install $PIP_OPTS 'numpy<2' --force-reinstall

Loading…
Cancel
Save