search

boostcampaitech3 / level2-data-annotation_cv-level2-cv-09

level2-data-annotation_cv-level2-cv-09 created by GitHub Classroom
0 stars 4 forks source link

[Discussion] .pth 이름 어떻게 저장할 지? #43

Open tjrudrnr2 opened 2 years ago

tjrudrnr2 commented 2 years ago

Why?

현재 latest.pth best_mean_loss.pth 이렇게 저장이 되는데 inference.py에 디폴트로 latest.pth로 경로가 되어있어서 제출 기회를 날릴 우려가 있음

How?

  1. latest.pth 저장 대신 real-latest.pth로 저장해서 위의 경우와 같이 제출 시 에러가 뜨게 만들기
  2. 헷갈릴 수도 있으니까 기존 대로 하기 => inference에서 그때 그때 .pth 경로 확인
  3. 다른 방법?
jeongjae96 commented 2 years ago

일단 기존 방식으로 latest.pth만 저장하다가 나중에 validation을 도입할 때, latest.pth (valid best_mean_loss)와 real_latest.path 두 개 구분해서 저장하는 것 어떨까요? 제가 질문을 잘 이해한건지는 잘 모르겠습니다.

yoonghee commented 2 years ago

기존 default baseline 코드는 마지막 모델만 latest.pth로 저장되어 있어 loss값이 가장 좋은 모델을 저장하고 싶어 best_mean_loss라는 이름으로 저장되게 만들었었습니다.

하지만 제출할 때 latest.pth로 제출해야하는 점을 감안해서 제출할 모델을 latest.pth로 변경해야하는데요 -- 기존 -- latest.pth & best_mean_loss.pth -> 제출한 pth파일을 latest.pth로 변경하여 서버제출

-- 변경 -- real-latest.pth & best_mean_loss.pth -> 제출할 pth파일을 latest.pth로 변경하여 서버제출

jeongjae96 commented 2 years ago

변경된 방법이 안전해 보이긴 하네요!

updaun commented 2 years ago

저는 object detection 할 때처럼 wandb 실험 명칭에 따라 폴더가 생성되도록 하면 좋겠어요.

혹시나 실수로 학습을 돌렸을 때 기존 pth 파일이 덮어쓰기 되는 현상도 같이 방지 했으면 좋겠습니다!

폴더가 이미 있으면 뒤에 넘버링 붙여서 새로운 폴더 생성 되면 좋겠습니다!!

km9mn commented 2 years ago

--model_dir을 바꾸면 새로운 폴더에 저장되긴 하는데 자동화하면 실수에 대비할 수 있을거 같아요