我正在尝试运行TF SLIM中的一个教程,在这个教程中,您可以使用Inception-V3 (~104Mb)对flowers数据集进行微调。GPU大约有2 2Gb的内存。当批处理大小超过8个时,我会收到一个错误,因为GPU内存不足。事实上,我似乎收到了几条消息,每条消息看起来都是这样的:
W tensorflow/core/common_runtime/bfc_allocator.cc:217] Ran out of memory trying to allocate 646.50MiB. The caller indicates that this is not a failure, but
TF-Slim使用自己的训练循环。然而,我想使用TF-Slim模型(Resnet50),同时仍然使用我自己的tensorflow训练循环。TF-Slim模型简单地输出预测,然后我计算我自己的总损失。但我想确保我没有错误地使用TF-Slim。TF-Slim训练过程如下所示:update_ops
# are run
我使用DatasetDataProvider从tfrecord获取图像。我可以‘打印(图像)’,但当使用‘sess.run(图像)’来获取它时,程序似乎陷入了一个无限循环。我不知道我是否犯了错误。 Tensor("Reshape_3:0", shape=(28, 28, 1), dtype=uint8, device=/device:CPU:0) from __future__ import absolute_import from __future__ i