Update README.md
Browse files
README.md
CHANGED
@@ -1,4 +1,18 @@
|
|
1 |
---
|
2 |
base_model:
|
3 |
- meta-llama/Llama-3.2-3B-Instruct
|
4 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
---
|
2 |
base_model:
|
3 |
- meta-llama/Llama-3.2-3B-Instruct
|
4 |
+
license: llama3.2
|
5 |
+
---
|
6 |
+
|
7 |
+
This repo contains RiskAgent model finetuned from meta-llama/Llama-3.2-3B-Instruct model using risk prediction data. More detail can be found from the [Github](https://github.com/AI-in-Health/RiskAgent)
|
8 |
+
|
9 |
+
The model was trained with the following hypermeters:
|
10 |
+
|
11 |
+
- cutoff_len: 3096
|
12 |
+
- num_train_epochs: 5
|
13 |
+
- learning_rate: 1.0e-4
|
14 |
+
- warnup_ratio: 0.1
|
15 |
+
- batch_size: 32
|
16 |
+
- finetuning_type: lora
|
17 |
+
|
18 |
+
Prior to utilizing our model, please ensure you have obtained the Llama licensing and access rights to the Llama model.
|