关注
看起来你的问题主要是在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,来提高模型的推理速度。
查看原帖
点赞 评论
相关推荐

点赞 评论 收藏
分享
牛客热帖
更多
正在热议
更多
# 为了求职,我做过的疯狂伪装 #
8141次浏览 113人参与
# 从顶到拉给所有面过的公司评分 #
5220次浏览 64人参与
# 职场破冰,你们都聊什么? #
4145次浏览 53人参与
# 晒晒你的中秋福利 #
13874次浏览 83人参与
# 聊聊这家公司值得去吗 #
547713次浏览 3657人参与
# 广联达求职进展汇总 #
9961次浏览 50人参与
# 产品每日一题 #
59985次浏览 604人参与
# 实习要如何选择和准备? #
113742次浏览 1435人参与
# 机械人,你最希望上岸的公司是? #
181159次浏览 1894人参与
# 你面试被问到过哪些不会的问题? #
15282次浏览 641人参与
# 机械笔面试考察这些知识点 #
9668次浏览 89人参与
# 如何快速融入团队? #
31435次浏览 278人参与
# bilibili求职进展汇总 #
80641次浏览 748人参与
# 浪潮求职进展汇总 #
15818次浏览 131人参与
# 你觉得早上几点上班合适? #
81622次浏览 329人参与
# 上班后和你想的一样吗? #
78404次浏览 628人参与
# 工作压力大怎么缓解 #
103972次浏览 1043人参与
# 远景求职进展汇总 #
60628次浏览 317人参与
# 电网笔面经互助 #
45722次浏览 428人参与
# 秋招报数:你投了多少家公司? #
22367次浏览 231人参与