mnist訓練現存不足報告ミス:Ran out of memory trying to allocate 957.03 MiB.See logs for memory state. W tensorfl
35703 ワード
参照先:https://stackoverflow.com/questions/39076388/tensorflow-deep-mnist-resource-exhausted-oom-when-allocating-tensor-with-shape
トレーニングmnistデータセット、テスト時にエラー、表示不足
解決策:
(1)
(2)に変更
トレーニングmnistデータセット、テスト時にエラー、表示不足
解決策:
(1)
print("test accuracy %g"%accuracy.eval(feed_dict={ x: mnist.test.images, y_: mnist.test.labels, keep_prob: 1.0}))
:
for i in xrange(10): testSet = mnist.test.next_batch(50) print("test accuracy %g"%accuracy.eval(feed_dict={ x: testSet[0], y_: testSet[1], keep_prob: 1.0}))
(2)に変更
accuracy_sum = tf.reduce_sum(tf.cast(correct_prediction, tf.float32))
good = 0
total = 0
for i in xrange(10):
testSet = mnist.test.next_batch(50)
good += accuracy_sum.eval(feed_dict={ x: testSet[0], y_: testSet[1], keep_prob: 1.0})
total += testSet[0].shape[0]
print("test accuracy %g"%(good/total))