pytorch模型的保存与加载 细节

关于torch的模型与保存还是很简单的
但是有些模型涉及到自定义参数,对于自定义的参数
只要是model=class SPDNetwork(torch.nn.Module):
这种 里面的自定义参数都会自动保存的
可以通过调式,debug 查看里面的变量
直接对model进行evaluate查看

突然想到上面的原因,准备再次记录下
为什么不用初始化上面的三个正交权重和fc权重呢,
看下面这两行代码

model = model.SPDNetwork()

model = torch.load(“C:\Users\L\Downloads\jieya\SPDNet-master\tmp\afew\spdnet_133c.model”)

第一行加粗的我注释掉了,如果直接使用torch.load()相当于没有初始化SPDNet的构造函数,也就是没有调用__init__()方法,而是直接使用了forward传输方法,关于怎么发现的是因为,我在SPDNet类的初始化增加了一个卷积,但是使用的时候说没有这个属性,然后突然想到这一点,特此记录

你可能感兴趣的:(pytorch,pytorch,深度学习,人工智能,python)