TensorFlow Keras:`validate_indices` 参数无效。索引总是在 CPU 上验证,从不在 GPU 上验证

Eug*_*ini 6 python conv-neural-network keras tensorflow

那是昨天,我正在训练模型。一切看起来都很好。今天 27/05/2021 我收到这个警告:

警告:tensorflow:来自/usr/local/lib/python3.7/dist-packages/tensorflow/python/ops/array_ops.py:5049:使用validate_indices调用gather(来自tensorflow.python.ops.array_ops)已被弃用,并将在以后的版本中删除。更新说明:该validate_indices参数无效。索引始终在 CPU 上验证,而从不在 GPU 上验证。

我在 GoogleColab pro 上运行我的 CNN。

有人知道该怎么做才能解决吗?我没有改变任何东西,我认为 Tensorflow-Keras 应该处理 CPU-GPU 计算。

之后,model.fit的正常流程:

纪元1/200

216/216 - 18s - 损失:...

纪元 2/200

216/216 - 2s - 损失:...