Feat: Weight save

[juyoung-hong]

Overview

• epoch ensemble 등 제출 용이성을 위해 model weight 저장 코드 변경
• model weight 저장 경로 변경에 따라 inference code도 동시 수정
• 새로운 변수 (save_every_epoch)가 새로 생김에 따라 configs/training.yaml 파일 수정

Change Log

• [x] saved/exp_name 폴더 자동생성 후, 해당 폴더 내에 model weight 저장 기능 구현
• [x] save_every_epoch 변수(bool) 값에 따라 epoch마다 model weight 저장 기능 구현
• [x] val_every 변수 값에 따라 validation 기준 best model weight 저장 기능 구현 (loss기준: 최소 loss)
• [x] val_every 변수 값에 따라 validation 기준 best model weight 저장 기능 구현 (metric 기준: mIoU)
• [x] inference.py 파일에서 model weight 경로 선택으로 코드 변경

To Reviewer

train시 아래 명령어로 테스트 할 수 있습니다.

python src/train.py name='test'

inference시 아래 명령어로 테스트 할 수 있습니다. 이전에 비해 달라진 점은 name이 아닌 weight라는 인자를 주어야하며, 이때 saved는 제외하고 saved폴더 안에 있는 test/best_mIoU.pt를 선택하여 주어야 합니다.

saved
 ├─test
    ├─best_IoU.pt
    ├─best_Ioss.pt
    ├─epoch_0.pt
    └─epoch_1.pt

python src/inference.py weight='test/best_mIoU.pt'

결과적으로 submission 폴더 내에 'test/best_mIoU.csv' 파일이 생성됩니다.

Issue Tags

2

[GeonHyeock]

정상적으로 작동하는것을 확인했습니다.!!