Numpy Array 保存与读取

保存数据预处理操作，在训练时可直接读取，节省预处理时间

import os
...
if not os.path.exists('filename'):
    # 预处理
    np.savez('filename.npz',x_array=x_array,y_array=y_array)
else:
    filename=np.load('filename')
    x_array=filename['x_array']
    y_array=filename['y_array']

x_array=torch.Tensor(x_array)
y_array=torch.Tensor(y_array)

Numpy Array 保存与读取

推荐阅读