ValueErrorr：Graphdef不能大于2GB

Question

错误是由https://github.com/rajkumarcm15/Medical-Image-Segmentation/blob/master/PixelWise/train_class.py#L254抛出的地方，我在保存训练后的模型。我该如何克服这一点？谢谢

Answer 1

您的训练循环中的以下内容将在的每次迭代上创建新的整形操作。所以如果你运行1000次迭代，你将有2000个重塑操作被添加到图表中，这可以很快地使图表定义超过2Gb限制（不幸的是，这个大小限制来自Protobufs并且不能轻易增加）。尝试在循环之外移动这些操作（以及任何其他操作）。

train_data = sess.run(tf.reshape(train_data, [-1, 1, _x_, _y_, _z_])) train_targ = sess.run(tf.reshape(train_targ, [-1, 1, 3, 1]))