vibhorag101
commited on
Commit
•
42bd259
1
Parent(s):
e9b8994
Update README.md
Browse files
README.md
CHANGED
@@ -21,7 +21,7 @@ pipeline_tag: text-generation
|
|
21 |
- 128GB RAM.
|
22 |
### Model Hyperparameters
|
23 |
- This [training script](https://github.com/vibhorag101/llama2-mental-therapy/blob/main/finetuneModel/finetuneScript.ipynb) was used to do the finetuning.
|
24 |
-
- The shareGPT format dataset was converted to llama-2 training format using this [script](https://github.com/vibhorag101/
|
25 |
- num_train_epochs = 3
|
26 |
- per_device_train_batch_size = 2
|
27 |
- per_device_eval_batch_size = 2
|
|
|
21 |
- 128GB RAM.
|
22 |
### Model Hyperparameters
|
23 |
- This [training script](https://github.com/vibhorag101/llama2-mental-therapy/blob/main/finetuneModel/finetuneScript.ipynb) was used to do the finetuning.
|
24 |
+
- The shareGPT format dataset was converted to llama-2 training format using this [script](https://github.com/vibhorag101/llama2-mental-therapy/blob/main/finetuneModel/llamaDataMaker.ipynb).
|
25 |
- num_train_epochs = 3
|
26 |
- per_device_train_batch_size = 2
|
27 |
- per_device_eval_batch_size = 2
|