boostcampaitech2 / klue-level2-nlp-02

klue-level2-nlp-02 created by GitHub Classroom
0 stars 6 forks source link

inference 실행시 checkpoint files 리스트업 기능 추가 #6

Closed Yebin46 closed 3 years ago

Yebin46 commented 3 years ago

inference.py, train.py, gitignore 수정하였습니다!

참고 이미지 image

  1. inference.py 실행시 자동으로 checkpoint들을 불러와 선택할 수 있도록 하였습니다. (참고 이미지 주황색 박스 참고)

  2. 1을 수행할 수 있도록 train.py 내에서 모델 저장시 폴더 저장 이름을 수정하였습니다. (기존 train.py에서 best_model이란 이름을 args.PLM-args.wandb_unique_tag 형태의 이름으로 바꾸었습니다.) 이 부분은 간단하기도 하고 직접 실행시켜본 게 아니라 꼭 한번 확인 부탁드립니다ㅠㅠ

  3. train.py에서 (classifier까지 학습된) pretrained model을 불러올 때 사이즈가 맞지 않은 경우 무시하고 불러올 수 있도록 ignore_mismatched_size 옵션을 추가했습니다.

  4. gitignore 파일을 제외한 숨김파일들을 업로드하지 않도록 바꾸었습니다. (만약 그래도 업로드가 된다면, git cache 문제이니 아래 코드를 실행해주세요.)

git rm -r --cached .
git add .
git commit -m "fixed untracked files"

감사합니다!

sangHa0411 commented 3 years ago

좋은 내용 너무 감사드립니다!

presto105 commented 3 years ago

감사합니다!