et al/AI Tech 6
mmdetection에 wandb 연결
코딩상륙작전
2024. 1. 11. 03:01
configs/_base_/default_runtime.py
위 파일 경로에서 vis_backends를 다음과 같이 고친다.
mm환경설정 :
pip install -U openmim
mim install mmengine
mim install "mmcv>=2.0.0"
vis_backends = [dict(type='LocalVisBackend'),
#dict(type='TensorboardVisBackend'), tensorboard가 default이다.
dict(type='WandbVisBackend',
init_kwargs={
'project': 'recycle_detection',
'group': 'sparse-rcnn' # wandibi에 그룹으로 저장된다.
})]
설치 후에 완디비 api를 넣어준다.