Lua 中写入文件很慢

我正在尝试在 Lua 中将一些数据写入文本文件,但是出于某种原因,它非常慢(仅 100 行需要 8 秒)。超过 100 行会进一步减慢,超过 1000 行,我的 GPU 就会耗尽内存。我做错了什么,请问?我需要在某个地方添加 collectgarbage() 吗?谢谢。

fp = io.open(('predictions/valid_test.txt'), "w")
local soft = nn.SoftMax()
local n = 100

for i = 1, n do
   local t_data = data[{{start_idx, end_idx},{},{},{}}]
   local outputs = model:forward(t_data):float()
   local soft_data = soft:forward(outputs)
   fp:write(fname[i] .. ',' ..  soft_data[1] .. ',' .. '\n')
   xlua.progress(i, n)
end
fp.close()
点赞
用户4013781
用户4013781

将下面翻译成中文并且保留原本的 markdown 格式

The error was actually a memory leak in Torch. Not sure why this was downvoted though, I wasn't implying that Lua is deficient! ;o

实际上,错误是Torch中的内存泄漏。不确定为什么会被踩,我并没有暗示Lua有缺陷! ;o

2016-02-05 23:42:21