随笔 - 204,  文章 - 0,  评论 - 19,  阅读 - 49万
01 2025 档案
多卡蒸馏模型
摘要:背景 需要对3B模型进行蒸馏,一张4090的卡无法完成实验。完成这个实验的前提是需要两张卡,一张用来加载学生模型,一张用来加载教师模型。 多卡使用 这里的多卡使用并不是像以往的方式,使用dataloaderparallel等方式,这种是数据并行的策略,不适合蒸馏的场景,因为蒸馏是一个模型做推理,一个 阅读全文
posted @ 2025-01-20 14:43 蔚蓝色の天空 阅读(16) 评论(0) 推荐(0) 编辑

< 2025年3月 >
23 24 25 26 27 28 1
2 3 4 5 6 7 8
9 10 11 12 13 14 15
16 17 18 19 20 21 22
23 24 25 26 27 28 29
30 31 1 2 3 4 5

点击右上角即可分享
微信分享提示