Fine-tuning a model on the KLUE-STS

Coment

가짜연구소 3기 KLUE로 모델 평가하기 4주차에서 발표한 자료입니다.

지난번 YNAT에 이어 STS에 대한 베이스라인 코드입니다.

KLUE에서 공개한 Pre-trained 모델의 weights를 불러와서 한국어 벤치마크 데이터셋 KLUE의  문장 유사도 평가 데이터셋 STS에 Fine-tuning하는 예제입니다.

 

Contents

• HuggingFace Datasets을 활용하여 KLUE 데이터셋 쉽게 전처리하기
• HuggingFace Hub에서 사전학습된 언어 모델을 다운로드 받아 사용하고, 학습한 모델을 업로드하여 공유하기
• Trainer 객체를 사용하여 모델 학습 및 평가 & hyperparameter search하기
• Weights & Biases를 활용하여 실험 관리하기

https://colab.research.google.com/drive/1uE7GrsZl8XrImf74bRoxRVYzoUfvMNzl?usp=sharing

Fine-tuning a model on the STS Task(v0.9).ipynb