Beomi / KcBERT

🤗 Pretrained BERT model & WordPiece tokenizer trained on Korean Comments 한국어 댓글로 프리트레이닝한 BERT 모델과 데이터셋

Home Page:https://huggingface.co/beomi/kcbert-base

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

안녕하세요, colab코드에 대해 질문이 있습니다.

KFromage opened this issue · comments

먼저 좋은 자료 공유해 주셔서 감사합니다.
저는 아직 초보지만 딥러닝에 관심이 있는 학생입니다.

colab에서 불러오는 dataset만 제가 원하는 데이터로 수정하여 학습을 진행해 봤습니다.

처음에는, colab의 KcBERT Large에서 pretrained_model만 beomi/kcbert-base로 수정해 시도해 봤는데,
학습이 완료된 모델이 세션 저장소에서 보이지 않았습니다.

두 번째에는, colab의 kcbert-nsmc에서 dataset만 바꾸어 진행해 봤는데, Exception: Model doesn't exists! Train first!
가 나왔습니다. 또한 predict.py를 실행했을 때에는 FileNotFoundError: [Errno 2] No such file or directory: './model/training_args.bin'이 나왔습니다.

colab에서 predict.py를 실행할 때 제가 준비해야 하는 다른 무언가가 있나요?
colab에서 제가 가진 데이터로 학습한 모델을 사용하거나, 로컬에서 사용하는 방법이 궁금합니다.