Pytorch 使用笔记_6.3

记录一下使用Pytorch的一些技巧

1. 首先是变量的数据类型（dtype）转换， 在信号处理中，因为经常会从mat文件中导入数据，在用scipy包的时候导入的一般都是ndarray类型，也就是numpy类型。numpy中数据类型转换可以通过astype(’ ')成员函数来进行转换。

 a  = numpy.array([1,2]
 a.astype('float32')

在torch中可以用type成员函数

b = torch.Tensor(2,2)
b.type(torch.float)

2. 然后是一些快捷键的问题。比如批量替换变量名，在写代码时通常会遇到改变变量名的时候，但有时候需要改变的变量出现的次数太多，需要批量操作。可以使用CRTL+R 快捷键 打开查找替换窗口，在左边的第一行输入你要查找的变量，下一行输入新变量名，再点击replace all就行。还有一些快捷键，比如说往右缩进用Tab键，往左缩进用shift+tab键。注释可以使用ctrl+/ , 取消注释也是一样，
   注释多行也可以使用’’’ ‘’'格式。还有具有强迫症的人写完一行代码可以使用ctrl + alt+L 键就可以让代码自动排版。

3. 画图我们经常会需要保存跑出来的结果图，可以使用下面代码

import time,os
time1=time.strftime('%Y-%m-%d') # read data today
sv_path=r'F:\UserOfLab\LJ\PythonCode\LISTA\Figure\1DResult'+time1 # Appoint figure path
os.makedirs(sv_path,exist_ok=True)
plt.savefig(f'{sv_path}/1DTestLoss_AddNoise.png')#保存文件在指定文件夹下很方便

第一行导入两个需要的包，第二行是读取当前日期，第三行是写明存放地址并带上了第二行生成的日期，第四行在第三行定义的路径里生成了文件夹，第五行保存图片在第四行生成的文件夹里并保存为PNG格式，当然可以改成其他，比如JPG, BMP等等。
这里说明一些细节，在第二行里开头加入r的原理是因为 \ 在python 里还可以被用来定义转义符号，这里加上r向编译器说明这里就是原来用法，或者说可以把 \ 改成 \ 或者 /。plt.show()一定要放在这些代码的后面，先show的话会使得保存空白图片, 下面再附上读取PNG的链接吧。
实现图片读取并显示

4.我们还会需要经常保存训练好的模型变量，代码如下，这里实际上保存是模型的参数

torch.save(model.state_dict(), "model_1D_AddNoise.pth")

读取模型参数

model = LISTANetwork(We=We, Len=Len, Max_iter=6, L=lambd, theta=theta).to(device)
pathfile = r'F:\UserOfLab\LJ\PythonCode\LISTA\model_1D.pth'
model.load_state_dict(torch.load(f=pathfile, map_location=device))