Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

model weight 저장 기능 고도화 #2

Open
5 tasks done
juyoung-hong opened this issue Dec 22, 2022 · 0 comments
Open
5 tasks done

model weight 저장 기능 고도화 #2

juyoung-hong opened this issue Dec 22, 2022 · 0 comments
Labels
enhancement New feature or request

Comments

@juyoung-hong
Copy link
Contributor

juyoung-hong commented Dec 22, 2022

Background

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

ToDo

  • saved/exp_name 폴더 자동생성 후, 해당 폴더 내에 model weight 저장 기능 구현
  • save_every_epoch 변수(bool) 값에 따라 epoch마다 model weight 저장 기능 구현
  • val_every 변수 값에 따라 validation 기준 best model weight 저장 기능 구현 (loss기준: 최소 loss)
  • val_every 변수 값에 따라 validation 기준 best model weight 저장 기능 구현 (metric 기준: mIoU)
  • inference.py 파일에서 model weight 경로 선택으로 코드 변경
@juyoung-hong juyoung-hong added the enhancement New feature or request label Dec 22, 2022
@juyoung-hong juyoung-hong mentioned this issue Dec 22, 2022
5 tasks
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancement New feature or request
Projects
None yet
Development

When branches are created from issues, their pull requests are automatically linked.

1 participant