분석시각화 대회 코드 공유 게시물은
내용 확인 후
좋아요(투표) 가능합니다.
허깅페이스 학습 파이프라인(klue/roberta-base, LB 0.805)
klue/bert-base를 기반으로 분류(contradiction, neutral, entailment)를 예측하는 코드입니다. 코드를 huggingface라이브러리 기반으로 작성하여, 비교적 간단하게 구현하였습니다. 간단한 분류 모델을 학습하는 데에 도움이 되었으면 좋겠습니다.
감사합니다:)
감사합니다! lr같은 하이퍼파라미터는 직접 조정 못하는 건가요??
TrainingArguments를 기반으로 조정 가능합니다.
좋은 글 감사합니다.
혹시 해당 코드는 로컬에서 돌리신걸까요?
구글 colab에서 돌리니 "CUDA error: device-side assert triggered" 에러가 발생해서요 ㅠ
코랩에서 돌렸고, 해당 에러는 입출력 처리가 잘못되어서 발생하는 에러인데, 여기서 제공하는 것 그대로 사용하시면 에러가 없을 듯 합니다...ㅠ
데이콘(주) | 대표 김국진 | 699-81-01021
통신판매업 신고번호: 제 2021-서울영등포-1704호
서울특별시 영등포구 은행로 3 익스콘벤처타워 901호
이메일 dacon@dacon.io | 전화번호: 070-4102-0545
Copyright ⓒ DACON Inc. All rights reserved
감사합니다!! XD