首页 Paddle框架 帖子详情
Paddle2.1在AIStudio跑知识蒸馏的问题
收藏
快速回复
Paddle框架 问答模型训练 488 2
Paddle2.1在AIStudio跑知识蒸馏的问题
收藏
快速回复
Paddle框架 问答模型训练 488 2

跑了一个知识蒸馏,然后知识蒸馏在AIStudioGPU环境跑的时候,蒸馏后的模型大小没有发生变化。

但是在本地WinodowsGPU环境下,模型大小由32M缩小为了20M,windowsCPU环境下模型大小更是32M缩小为了15M.不知道哪里有问题。

0
收藏
回复
全部评论(2)
时间顺序
三岁
#2 回复于2021-07

这个会不会和GPU的,cpu的显存什么的存在关系,可以看一下最后的效果,应该是不一样的

0
回复
Medlen
#3 回复于2021-07

其实这个应该是可以比较一下参数的。 用model.state_dict() 保存模型, 然后读取两个不同大小的模型,打印出来整个模型所有的层,和层的shape,看看是否是某些层参数减少了。如果模型参数量是一致的,那我觉得可能要考虑是不是不同环境用的压缩算法啥的不一样...

0
回复
需求/bug反馈?一键提issue告诉我们
发现bug?如果您知道修复办法,欢迎提pr直接参与建设飞桨~
在@后输入用户全名并按空格结束,可艾特全站任一用户