Open
Description
In the gsm8k script (link), the distilled dataset is generated using fp16 precision, while the model is trained on this dataset using bf16.
Shouldn't the precision format be consistent throughout the process?
Generate distilled dataset using fp16: [Line 35](https://github.com/sail-sg/sdft/blob/bfb6c255fccdce7459235c20f19a3b9817a9cd5d/scripts/gsm8k/sdft.sh#L35)
Train on distilled dataset using bf16: [Line 61](https://github.com/sail-sg/sdft/blob/bfb6c255fccdce7459235c20f19a3b9817a9cd5d/scripts/gsm8k/sdft.sh#L61)
Metadata
Metadata
Assignees
Labels
No labels