2016-02-05 39 views
-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() 
+1

只是評論'fp:寫('線和運行,看它的文件寫入是否很慢 – moteus

+2

Lua可以很快寫入文件,我相信你遇到的問題是緩慢的是其他線你有沒有測試過他們的速度? – warspyking

回答

0

該錯誤實際上是火炬內存泄漏。不知道爲什麼這是downvoted,但我並不是暗示Lua是有缺陷的! ; 0