File size: 691 Bytes
0568789
 
 
 
 
a7be7fd
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
---
license: other
license_name: yi-license
license_link: LICENSE
---
About 961 steps total, Yi-34B-200K llamafied DPO trained for 1 epoch on rawrr_v2 dataset via unsloth at prompt length of 400 and max length of 700, lr 0.000045 \
Model initialized with max_positional_embeddings of 4096 to not OOM. \
Training done on RTX 3090 Ti in about 14 hours. \
Average mem usage was like 23.89 / 23.99 GiB, so very close to OOM at all times. \
I trained it with XFCE on one 1080p monitor loaded up, on more fancy DM it would probably OOM with the same setup. \
I am not sure what's the purpose of max_prompt_length being separate from max_length, so I may have used it wrong, I should read up on it.