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']
Tags: