nenad1002 commited on
Commit
f7d70ff
1 Parent(s): 393d816

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -89,7 +89,7 @@ For ReFT, the nodes in the last 8 layers were unfrozen with attention to allow t
89
  After 3 to 4 epochs, the model began to overfit regardless of the strategies employed. Increasing both batch size and the number of epochs resulted in higher final training and evaluation cross-entropy.
90
 
91
  Following an extensive grid search, supervised fine-tuning of [Llama 3.1-8B-Instruct](https://huggingface.co/meta-llama/Meta-Llama-3.1-8B-Instruct) with LoRA+ and the parameters mentioned below yielded the best training and evaluation cross-entropy.
92
- I've chosen the size ratio between the matrices A and B of 8. The matrix A weights were initialized using the He method, while the matrix B values started with zero. A Gaussian initialization of weights was also considered, but led to a suboptimal result.
93
  #### Preprocessing [optional]
94
 
95
  [Coming soon]
 
89
  After 3 to 4 epochs, the model began to overfit regardless of the strategies employed. Increasing both batch size and the number of epochs resulted in higher final training and evaluation cross-entropy.
90
 
91
  Following an extensive grid search, supervised fine-tuning of [Llama 3.1-8B-Instruct](https://huggingface.co/meta-llama/Meta-Llama-3.1-8B-Instruct) with LoRA+ and the parameters mentioned below yielded the best training and evaluation cross-entropy.
92
+ I've chosen the size ratio between the matrices A and B of 8. The matrix A weights were initialized using the He method, while the matrix B values started with zero. Different Gaussian initialization of weights were also considered, but led to a suboptimal result. Since a custome optimizer was built here, I will share the optimizer code on my private GitHub account soon.
93
  #### Preprocessing [optional]
94
 
95
  [Coming soon]