关注
看起来你的问题主要是在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,来提高模型的推理速度。
查看原帖
点赞 评论
相关推荐
06-02 23:35
门头沟学院 后端 
点赞 评论 收藏
分享

点赞 评论 收藏
分享
牛客热帖
更多
正在热议
更多
# 实习生的蛐蛐区 #
55050次浏览 411人参与
# 运营面经 #
115649次浏览 1250人参与
# 你认为小厂实习有用吗? #
20886次浏览 250人参与
# 说说你知道的学历厂 #
39576次浏览 238人参与
# 应届生,你找到工作了吗 #
21386次浏览 152人参与
# 三一重工求职进展汇总 #
13379次浏览 61人参与
# 材料人,你们签了哪个公司 #
7490次浏览 18人参与
# 计算机有哪些岗位值得去? #
17126次浏览 160人参与
# 哪一瞬间觉得自己长大了 #
10072次浏览 228人参与
# 面试尴尬现场 #
32646次浏览 217人参与
# 你找工作的时候用AI吗? #
18915次浏览 232人参与
# 下班后的时间你怎么安排 #
10368次浏览 140人参与
# 烟草笔面经互助 #
17854次浏览 184人参与
# 秋招最大的收获是什么? #
36116次浏览 309人参与
# 社会教会你的第一课 #
36885次浏览 463人参与
# 电网笔面经互助 #
36918次浏览 357人参与
# 硬件应届生薪资是否普遍偏低? #
75431次浏览 520人参与
# lastday知无不言 #
58335次浏览 475人参与
# 你的领导最像哪种动物,为什么? #
14379次浏览 107人参与
# 学历贬值真的很严重吗? #
22443次浏览 163人参与