ysakuramoto
commited on
Commit
•
3ec94b9
1
Parent(s):
00cfab8
Update README.md
Browse files
README.md
CHANGED
@@ -49,7 +49,7 @@ model = MobileBertForSequenceClassification.from_pretrained("ysakuramoto/mobileb
|
|
49 |
- 条件
|
50 |
- ストックーマークさんのwikipediaデータセットで学習・推論。(https://github.com/stockmarkteam/ner-wikipedia-dataset)
|
51 |
- BERTモデルに、'cl-tohoku/bert-base-japanese-whole-word-masking'を利用。
|
52 |
-
- 推論データ n=2,140。精度は完全一致の
|
53 |
- エポック数=10, lr=1e-4
|
54 |
- 推論時の高速化として、枝刈り・量子化・jitコンパイルを実施。
|
55 |
- Google Colabにて、学習にGPU、推論にCPUを利用。推論はバッチ処理でなく1件ずつ処理。
|
|
|
49 |
- 条件
|
50 |
- ストックーマークさんのwikipediaデータセットで学習・推論。(https://github.com/stockmarkteam/ner-wikipedia-dataset)
|
51 |
- BERTモデルに、'cl-tohoku/bert-base-japanese-whole-word-masking'を利用。
|
52 |
+
- 推論データ n=2,140。精度は完全一致のf-measure。
|
53 |
- エポック数=10, lr=1e-4
|
54 |
- 推論時の高速化として、枝刈り・量子化・jitコンパイルを実施。
|
55 |
- Google Colabにて、学習にGPU、推論にCPUを利用。推論はバッチ処理でなく1件ずつ処理。
|