-1
我正在嘗試將一些數據寫入Lua中的文本文件,但由於某種原因,它非常慢(僅爲100行8s)。超過100行的速度進一步下降,超過1000行,我的GPU耗盡內存。請問我哪裏錯了?我需要在某處添加collectgarbage()
嗎?謝謝。在Lua中緩慢寫入文件
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()
只是評論'fp:寫('線和運行,看它的文件寫入是否很慢 – moteus
Lua可以很快寫入文件,我相信你遇到的問題是緩慢的是其他線你有沒有測試過他們的速度? – warspyking