chainerでGPUメモリオーバー
chainerで下記のようなループを回していたら2度目以降にメモリアロケートエラーになってしまいました。
一度目のループでメモリが足りているのだから、メモリの開放かなにかをしないといけないのでしょうか?
for k in range( 0, len(dataset),batchsize ):
for j in range(batchsize):
if j+k >= len(dataset):
break
color , line = dataset.get_example(j+k)
_x[j,:] = color
x = cuda.to_gpu(_x)
y ,e = cnn.calc(Variable(x), test=True )
output = y.data.get()