Numpy Array save and read

Preprocessing operations save data can be read directly in training, time-saving pretreatment

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)