Update README.md
Browse files
README.md
CHANGED
@@ -10,7 +10,7 @@ This version of DNABERT2 has been changed to be able to output the attention too
|
|
10 |
|
11 |
**To the author of DNABERT2, feel free to use those modifications.**
|
12 |
|
13 |
-
Use ```--model_name_or_path jaandoui/
|
14 |
|
15 |
Most of the modifications were done in Bert_Layer.py.
|
16 |
It has been modified especially for fine tuning and hasn't been tried for pretraining.
|
|
|
10 |
|
11 |
**To the author of DNABERT2, feel free to use those modifications.**
|
12 |
|
13 |
+
Use ```--model_name_or_path jaandoui/DNABERT2-AttentionExtracted``` instead of the original repository to have access to the attention.
|
14 |
|
15 |
Most of the modifications were done in Bert_Layer.py.
|
16 |
It has been modified especially for fine tuning and hasn't been tried for pretraining.
|