Python最簡單的永久存儲:pickle模塊

現在深度學習很多使用的是動態圖,方便調試程序

但是隨之也帶來了一個問題,就是如果程序中途,我們關閉了運行環境

那再進入以後,中間變量就沒了,要從頭運行一遍

對於深度學習來說,很多數據的處理也能是批量進行,中間變量如果一旦損失,那非常浪費時間

那最好的方法就是把內存中的數據暫時存放到硬盤上,這樣就斷電也不丟失啦!

在 Python 中,雖然有很多存儲模塊,但是我覺得最簡單的就是 pickle 模塊了

程序案例:

import pickle
import numpy as np

a = np.array([[1,2,3],[4,5,6]])
b = 'www'
c = [a,b]

f = open('data.pk','wb')
pickle.dump(c, f)
f.close()

fr = open('data.pk','rb')
d = pickle.load(fr)
fr.close()
print(d)

運行結果:

[array([[1, 2, 3],
       [4, 5, 6]]), 'www']
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章