Jupyter Notebooks 预测总是相同的 class 与 GPU 和预训练的 Keras 检查点
[英]Jupyter Notebooks predicting always the same class with GPU and pretrained Keras checkpoint
问题描述
当我使用load_weights预测基于 VGG16 的预训练检查点时，我总是得到相同的 class 和 output 张量。 这只发生在我使用 GPU（我有 2 个 GPU）时。
另一方面，如果我只通过设置使用我的 CPU：
os.environ["CUDA_VISIBLE_DEVICES"] = "-1"
我得到了预期的结果。 这在 Jupyter Notebooks 中发生在我身上，但在 google colab 中不会发生。 在 google colab 中，无论如何我都能得到准确的结果。
- 我正在使用 Keras 2.4.3
- TensorFlow 2.4.1
- CUDA 11.2
0 个解决方案
Keras CNN-总是在平衡数据集中预测相同的类，但是准确性很高-为什么？
[英]Keras CNN - always predicting the same class in a balanced dataset, but the accuracy is high - why?
暹罗网络的 Keras 模型不学习并始终预测相同的输出
[英]Keras Model for Siamese Network not Learning and always predicting the same ouput
sklean SVM / SVC 总是为任何给定的输入预测相同的类
[英]sklean SVM / SVC always predicting the same class for any given input
Keras 图像分类网络总是预测一个 class，并保持在 50% 的准确率
[英]Keras image classification network always predicting one class, and stays at 50% accuracy
声明:本站的技术帖子网页，遵循CC BY-SA 4.0协议，如果您需要转载，请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.