inference.py 실행시 자동으로 checkpoint들을 불러와 선택할 수 있도록 하였습니다. (참고 이미지 주황색 박스 참고)
1을 수행할 수 있도록 train.py 내에서 모델 저장시 폴더 저장 이름을 수정하였습니다. (기존 train.py에서 best_model이란 이름을 args.PLM-args.wandb_unique_tag 형태의 이름으로 바꾸었습니다.) 이 부분은 간단하기도 하고 직접 실행시켜본 게 아니라 꼭 한번 확인 부탁드립니다ㅠㅠ
train.py에서 (classifier까지 학습된) pretrained model을 불러올 때 사이즈가 맞지 않은 경우 무시하고 불러올 수 있도록 ignore_mismatched_size 옵션을 추가했습니다.
gitignore 파일을 제외한 숨김파일들을 업로드하지 않도록 바꾸었습니다.
(만약 그래도 업로드가 된다면, git cache 문제이니 아래 코드를 실행해주세요.)
inference.py, train.py, gitignore 수정하였습니다!
참고 이미지
inference.py 실행시 자동으로 checkpoint들을 불러와 선택할 수 있도록 하였습니다. (참고 이미지 주황색 박스 참고)
1을 수행할 수 있도록 train.py 내에서 모델 저장시 폴더 저장 이름을 수정하였습니다. (기존 train.py에서 best_model이란 이름을 args.PLM-args.wandb_unique_tag 형태의 이름으로 바꾸었습니다.) 이 부분은 간단하기도 하고 직접 실행시켜본 게 아니라 꼭 한번 확인 부탁드립니다ㅠㅠ
train.py에서 (classifier까지 학습된) pretrained model을 불러올 때 사이즈가 맞지 않은 경우 무시하고 불러올 수 있도록 ignore_mismatched_size 옵션을 추가했습니다.
gitignore 파일을 제외한 숨김파일들을 업로드하지 않도록 바꾸었습니다. (만약 그래도 업로드가 된다면, git cache 문제이니 아래 코드를 실행해주세요.)
감사합니다!