unfortunately, I cant run 9b full fientuning - have access to 48GB gpus. I agree for the 4b + lora setup.
I haven't tried different lora configs (rank, alpha) or learning rates.
EMA gain shrinks with scale: one possible explanation for that: base is already quite strong and a thin lora
might not promote self-distillation that much. the paper if I remember correctly was giving +3 accuracy gap after 2 epochs of training.
the purpose of the repo was mostly to get lora+zero3 run rather than answer that kind of questions. would be great if people use it to answer them :)