关注
看起来你的问题主要是在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,来提高模型的推理速度。
查看原帖
点赞 评论
相关推荐
点赞 评论 收藏
分享
牛客热帖
更多
- 1... 学院本27届如何找到实习的一点点经验1.4W
- 2... 大二双非非科班成功拿下字节offer1.3W
- 3... 💬救命!原来被AI整破防了是这种感觉...1.2W
- 4... 那些在焦虑里种下的希望,终于在大厂 offer 里开了花1.1W
- 5... 面试时反问这些显得很有"水平"1.1W
- 6... 二战学校领导,**学校是人 ?阿里子公司不让去实习?9909
- 7... 从过来人视角告诉你,你不会找不到工作7706
- 8... 主包发现妈妈在玩xhs后7194
- 9... 为什么我怎么选都是错的6203
- 10... 谢谢ai告诉我,人生或许根本不需要宏大目标5278
正在热议
更多
# 牛油的搬砖plog #
2317次浏览 30人参与
# 安利/避雷我的专业 #
68522次浏览 501人参与
# 京东TGT #
57061次浏览 194人参与
# 一人一个landing小技巧 #
4610次浏览 90人参与
# 选完offer后,你后悔学本专业吗 #
40217次浏览 219人参与
# 面试问题记录 #
80964次浏览 1115人参与
# 工作中,你有没有遇到非常爱骂人的领导? #
22583次浏览 150人参与
# 面试经验谈 #
58039次浏览 926人参与
# 上班到公司第一件事做什么? #
42812次浏览 390人参与
# 技术转行的心路历程 #
49831次浏览 665人参与
# 找工作的破防时刻 #
3985次浏览 47人参与
# 实习生应该准时下班吗 #
205623次浏览 1337人参与
# 求职季如何保持心态不崩 #
110329次浏览 911人参与
# 工作时那些社死瞬间 #
28072次浏览 202人参与
# 互联网行业现在还值得去吗 #
8682次浏览 42人参与
# 你觉得技术面多长时间合理? #
88589次浏览 647人参与
# 硬件人的简历怎么写 #
259790次浏览 2906人参与
# 拼多多工作体验 #
19171次浏览 153人参与
# 面试吐槽bot #
21154次浏览 117人参与
# 工作一周年分享 #
21477次浏览 113人参与
# 你遇到过哪些神仙同事 #
77618次浏览 669人参与