Boostcamp 내에서 ImageNet-Sketch 데이터셋을 기반으로 한 이미지 분류 대회를 진행하였습니다.
apt-get update -y
apt-get install -y libgl1-mesa-glx libglib2.0-0
apt install -y git
git clone https://github.com/boostcampaitech7/level1-imageclassification-cv-02.git
cd level1-imageclassification-cv-02
이 프로젝트는 Poetry를 사용하여 의존성을 관리합니다. 설치 방법은 다음과 같습니다:
# Poetry 설치
python -m venv competition1
source competition1/bin/activate
pip install poetry
poetry lock
poetry install
- 추가로 설치한 패키지들도 있어서, pip도 해주어야 합니다. (lock으로 설정 X)
pip install -r requirements.txt
python train.py --config configs/train_configs/train/config.yaml
python test.py --config configs/train_configs/test/config.yaml
- Wandb: https://kr.wandb.ai/
터미널에서 wandb login
실행 후, 본인의 API를 입력합니다.
- Wandb 쓰는 방법 👉
--use_wandb
추가
python train.py --config configs/train_configs/train/config.yaml --use_wandb
- sweep을 사용할 때는, configs/train_configs/train/config.yaml 내에 use_sweep을 True로 변경합니다.
- agent 생성
wandb sweep configs/train_configs/train/sweep.yaml
- agent 적용
wandb agent <your wandb agent> -- count 5