kkmkorea commited on
Commit
86a2a73
·
1 Parent(s): 52be812

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +17 -1
README.md CHANGED
@@ -42,6 +42,22 @@ KorSciDeBERTa는 Microsoft DeBERTa 모델의 아키텍쳐를 기반으로, 논
42
 
43
  - Mecab 설치 참고: 다음 링크에서 '사용방법'. https://aida.kisti.re.kr/model/9bbabd2d-6ce8-44cc-b2a3-69578d23970a
44
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
45
  2. git clone https://huggingface.co/kisti/korscideberta; cd korscideberta
46
 
47
  - **korscideberta-abstractcls.ipynb**
@@ -52,7 +68,7 @@ from transformers import AutoModelForSequenceClassification
52
 
53
 
54
  tokenizer = DebertaV2Tokenizer.from_pretrained("kisti/korscideberta")
55
- model = AutoModelForSequenceClassification.from_pretrained("kisti/korscideberta", num_labels=6, hidden_dropout_prob=0.1, attention_probs_dropout_prob=0.1)
56
  #model = AutoModelForMaskedLM.from_pretrained("kisti/korscideberta")
57
  ''''''
58
  train_metrics = trainer.train().metrics
 
42
 
43
  - Mecab 설치 참고: 다음 링크에서 '사용방법'. https://aida.kisti.re.kr/model/9bbabd2d-6ce8-44cc-b2a3-69578d23970a
44
 
45
+ - Colab 사용하는 경우 Mecab 설치:
46
+
47
+ <pre><code>
48
+ ! git clone https://github.com/SOMJANG/Mecab-ko-for-Google-Colab.git
49
+ % cd Mecab-ko-for-Google-Colab/
50
+ ! bash install_mecab-ko_on_colab_light_220429.sh
51
+
52
+ </code></pre>
53
+
54
+ - ImportError: accelerate>=0.20.1 에러 발생시 해결법
55
+
56
+ !pip install -U accelerate; pip install -U transformers; pip install pydantic==1.8
57
+
58
+ : 설치 후 런타임 재시작
59
+
60
+
61
  2. git clone https://huggingface.co/kisti/korscideberta; cd korscideberta
62
 
63
  - **korscideberta-abstractcls.ipynb**
 
68
 
69
 
70
  tokenizer = DebertaV2Tokenizer.from_pretrained("kisti/korscideberta")
71
+ model = AutoModelForSequenceClassification.from_pretrained("kisti/korscideberta", num_labels=7, hidden_dropout_prob=0.1, attention_probs_dropout_prob=0.1)
72
  #model = AutoModelForMaskedLM.from_pretrained("kisti/korscideberta")
73
  ''''''
74
  train_metrics = trainer.train().metrics