关注
看起来你的问题主要是在GPU上运行模型时,输出结果全是0,而在CPU上运行则正常。这可能是由于GPU和CPU的浮点运算精度不同导致的。你可以尝试以下几种方法来解决这个问题:
1. 检查你的代码,确保在GPU和CPU上运行的代码是完全相同的。
2. 在GPU上运行时,尝试使用更低的浮点精度,例如half或float16。这可以通过在TensorFlow中设置session的配置来实现,例如:
```python
config = tf.ConfigProto()
config.gpu_options.per_process_gpu_memory_fraction = 0.5
config.gpu_options.allow_growth = True
sess = tf.Session(config=config)
```
3. 尝试使用不同的优化器或学习率,以避免梯度爆炸或消失。
4. 如果以上方法都无法解决问题,你可以尝试使用其他深度学习框架,如PyTorch,看看问题是否依然存在。
至于实时性问题,你可以尝试使用轻量级模型,如MobileNet或ShuffleNet,或者使用模型压缩技术,如知识蒸馏,来减小模型的大小和计算量。同时,你也可以尝试使用更快的推理引擎,如TensorRT或ONNX Runtime,来提高模型的推理速度。
查看原帖
点赞 评论
相关推荐
查看27道真题和解析 点赞 评论 收藏
分享
点赞 评论 收藏
分享
牛客热帖
更多
正在热议
更多
# 我的求职总结 #
476944次浏览 6752人参与
# AI让海力士市值突破9000亿美元 #
8650次浏览 105人参与
# 风评不好的公司,你会去吗? #
154883次浏览 705人参与
# 产品面经 #
297109次浏览 2216人参与
# 哪一瞬间让你觉得“这班不如不上” #
46554次浏览 278人参与
# 我想象的工作vs实际工作 #
709812次浏览 5053人参与
# 牛油的搬砖plog #
205248次浏览 1325人参与
# 什么专业适合考公 #
72015次浏览 426人参与
# 拼多多工作体验 #
61039次浏览 427人参与
# 得物app工作体验 #
67102次浏览 118人参与
# 这些公司卡简历很严格 #
106964次浏览 473人参与
# 职场新人体验 #
194930次浏览 1276人参与
# 产运销实习日记 #
106244次浏览 738人参与
# 小厂实习有必要去吗 #
93951次浏览 443人参与
# 百度工作体验 #
338955次浏览 2302人参与
# 春招什么时候投? #
56834次浏览 713人参与
# 重来一次,我还会选择这个专业吗 #
470312次浏览 4001人参与
# 你找工作的时候用AI吗? #
211894次浏览 1027人参与
# 为了求职,我做过的疯狂伪装 #
89000次浏览 785人参与
# 米哈游求职进展汇总 #
694725次浏览 3378人参与
