Update README.md

README.md

@@ -20,7 +20,27 @@ It achieves the following results on the evaluation set:
 
 ## Model description
 
-More information needed
+Trained on single ID token "5K dataset" filtered to 4k sequences (20% for test data)
+
+lora_alpha = 64 <--------------
+
+lora_dropout = 0.1
+
+lora_r = 64 <---------
+
+epochs = 3
+
+learning rate = 3e-4
+
+warmup_steps=500
+
+gradient_accumulation_steps = 1
+
+train_batch = 1
+
+eval_batch = 1
+
+ONLY ATTENTION LAYER <---------------------
 
 ## Intended uses & limitations