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