保存結果到文件中(Lua/Python)

在訓練網絡中經常需要保存一些參數或者結果,通常保存爲.log日誌文件或.txt文件。
保存到文件中,一般爲三個步驟:
1. 創建可寫文件,並打開
2. 序列化要保存的數據
3. 關閉文件

因爲會訓練多次,如果這些結果都想保存下來,那麼通常會獲取系統時間作爲文件名。
Note:因爲涉及到Torch(Lua)實現轉TensorFlow(Python)實現,這裏給出兩種語言的實現。

Lua:

require 'os'
localDate = os.date("%Y%m%d-%H%M%S")  --獲取系統時間格式爲:20180910-213256,即2018年09月10213256秒。
resultFd = io.open("./result/result_" .. localDate .. ".txt", "w")
for i = 1, Num do
    resultFd:write(tostring(output[i][1]) .. "\n")
end
resultFd:close()

Python2:

import time
localTime = time.strfTime('%Y%m%d-%H%M%S',time.localtime(time.time()))
resultFd = open('./result/result_' + localDate + '.txt', 'w')
for i in xrange(Num):
    resultFd.write(str(output[i,1]) + '\n')
resultFd.close()
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章