boostcampaitech3 / level2-semantic-segmentation-level2-cv-17

[2022.04.25 ~ 2022.05.12] Recycle Trash Semantic Segmentation Competition - 부스트캠프 AI Tech 3기

4 stars 2 forks source link

[Baseline] torch train.py save & wandb config #21

Closed Dongwoo-Im closed 2 years ago

Dongwoo-Im commented 2 years ago

Background

현재 모델 저장 시 latest, best_loss, best_miou 라는 이름으로 저장되고 있는데 조금 불편하다고 느꼈다. 또한 wandb config 저장이 안 되고 있다.

Content

작업 내용

[x] best_miou.pth, best_loss.pth 이름에 epoch 정보를 반영
[x] latest.pth도 지금처럼 하나만 저장하는 것이 아니라, save_interval 값마다 모델을 저장
[x] wandb config 기록
[x] optimizer.py scheduler.py 추가
[x] smp_model.py 다양한 모델 추가

작업 예정

train.py
wandb_setup.py (?)

Details

22 PR