我使用 slim 定义的 Inception V3 模型来预测自己的图像。模型变量从检查点文件中恢复。当我运行它时，我发现没有使用 GPU。我已经设置tf.device('/gpu:0')了，但它不起作用。所以，我很困惑为什么训练/重新训练模型可以在 GPU 上工作，但评估不能？我只是按照 inception_eval.py 中的代码进行操作。
在此处输入图像描述
问问题
246 次
我使用 slim 定义的 Inception V3 模型来预测自己的图像。模型变量从检查点文件中恢复。当我运行它时，我发现没有使用 GPU。我已经设置tf.device('/gpu:0')了，但它不起作用。所以，我很困惑为什么训练/重新训练模型可以在 GPU 上工作，但评估不能？我只是按照 inception_eval.py 中的代码进行操作。
在此处输入图像描述
