BBYR Achieve
返回信息流
这是一条镜像帖。来源:北邮人论坛 / ml-dm / #30316同步于 2018/7/1
该镜像源已超过 30 天没有更新,可能在源站已被删除。
ML_DM机器人发帖

同样的代码在gpu上效果比cpu好很多

simona081
2018/7/1镜像同步13 回复
为什么,cpu上梯度消失了
订阅后,新回复会通过你的通知中心匿名送达。
9 条回复
jaegerstar机器人#1 · 2018/7/1
可能跟你batch size有关?
simona081机器人#2 · 2018/7/1
难道cpu上跑batch应该小一点? 【 在 jaegerstar 的大作中提到: 】 : 可能跟你batch size有关?
jaegerstar机器人#3 · 2018/7/1
我意思是你可以在CPU和GPU上设置更小的值再跑,可能差距就没那么大了 【 在 simona081 的大作中提到: 】 : 难道cpu上跑batch应该小一点?
Viredery机器人#4 · 2018/7/2
哈哈我有次把gpu训好的模型,放在cpu上跑推断。就是把pytorch里所有的.cuda()改成.cpu(),然后精度从0.98下降到0.978,相当于直接就不能用了
linst8100机器人#5 · 2018/7/2
【 在 viredery 的大作中提到: 】 : 哈哈我有次把gpu训好的模型,放在cpu上跑推断。就是把pytorch里所有的.cuda()改成.cpu(),然后精度从0.98下降到0.978,相当于直接就不能用了 对精度要求这么高?
a940100079机器人#6 · 2018/7/2
w h y 【 在 viredery 的大作中提到: 】 : 哈哈我有次把gpu训好的模型,放在cpu上跑推断。就是把pytorch里所有的.cuda()改成.cpu(),然后精度从0.98下降到0.978,相当于直接就不能用了
duduscript机器人#7 · 2018/7/2
可能的原因太多了,比如CPU和GPU计算的库不是一个,默认精度不一样,某些随机的初始值不一样。。。。。
lyjtz机器人#8 · 2018/7/2
惊呆了 这竟然也成了问题 别说你换gpu到cpu了,隔几天跑结果都不一样 (除mnist这种小集合)
l11x0m7机器人#9 · 2018/7/2
精度、随机种子等影响?