[New Feature] 학습끝난 pth 불러와서 이어서 학습하기

boostcampaitech3 / level2-data-annotation_cv-level2-cv-15

level2-data-annotation_cv-level2-cv-15 created by GitHub Classroom

6 stars 1 forks source link

[New Feature] 학습끝난 pth 불러와서 이어서 학습하기 #20

Closed Lim-JH-Laskaris closed 2 years ago

Lim-JH-Laskaris commented 2 years ago

what?

학습끝난 pth 불러와서 이어서 학습할 수 있는 기능 추가
why?
큰 데이터셋으로 학습할 경우 200에폭에 3일 가까운 시간이 걸리기도 하는 바, 이를 활용하지 않고 처음부터 다시 학습하는 것은 시간 및 자원 낭비가 크다.
TODO
train.py에 model load 코드 추가
train.py에 aurg로 --load 추가

Lim-JH-Laskaris commented 2 years ago

--load 사용법 --load 디폴드 값은 F로, 이때는 처음부터 학습됩니다. --load T 로 주면, 정균님의 0.67모델을 불러옵니다.(단, trained_models 폴더에 해당 파일이 있어야 합니다.) --load L 로 주면, 기존에 학습하던 pth파일을 불러옵니다 exp_name이 동일해야 합니다. --load "~~~~.pth" 로 주면 해당 pth파일을 불러옵니다.

boostcampaitech3 / level2-data-annotation_cv-level2-cv-15

[New Feature] 학습끝난 pth 불러와서 이어서 학습하기 #20

what?

why?

TODO