https://github.com/rajkumarcm15/Medical-Image-Segmentation/blob/master/PixelWise/train_class.py#L254在训练后保存模型的位置引发错误。我怎么能克服这个？谢谢
答案 0 :(得分：0)
训练循环中的以下几行在每次迭代上创建新的重塑操作。因此，如果您运行1000次迭代，您将在图表中添加2000个重塑操作，这可以非常快速地使图形定义超过2Gb限制（不幸的是，此大小限制来自Protobufs并且不容易增加） 。尝试在循环外移动这些操作（以及其他任何tf。*）。
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]))