关注
看起来你的问题主要是在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,来提高模型的推理速度。
查看原帖
点赞 评论
相关推荐
查看15道真题和解析 点赞 评论 收藏
分享
牛客热帖
更多
正在热议
更多
# 春招/暑实第一面是哪家? #
4599次浏览 61人参与
# 面试官最爱问的 AI 问题是...... #
3362次浏览 117人参与
# 跟HR说什么能被秒回? #
2231次浏览 39人参与
# 机械制造面试点评 #
90280次浏览 482人参与
# 如何一边实习一边找下家? #
2644次浏览 42人参与
# 你收到了哪些公司的笔试? #
5755次浏览 25人参与
# 你的嫡系AI是哪个? #
1048次浏览 32人参与
# 你现在的工作,是“成长”还是“消耗”? #
4312次浏览 68人参与
# 现在入门AI应该走哪些方向? #
1014次浏览 27人参与
# 找不到好工作选择GAP真的丢人吗 #
102420次浏览 1020人参与
# 你认为小厂实习有用吗? #
127934次浏览 704人参与
# 金三银四,你的春招进行到哪个阶段了? #
19587次浏览 264人参与
# 滴滴笔试 #
38478次浏览 215人参与
# 追觅科技求职进展汇总 #
35739次浏览 190人参与
# 我的岗位说明书 #
316752次浏览 2784人参与
# 通信硬件公司爆料 #
200342次浏览 550人参与
# 哪些公司真双非友好? #
67410次浏览 281人参与
# 职场上哪些行为很加分? #
339722次浏览 3796人参与
# 实习进度记录 #
1219630次浏览 11864人参与
# 你上一次加班是什么时候? #
140073次浏览 780人参与