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]))