尝试在开源项目Hyperspectral-Classification Pytorch中加入自己的网络——记一次令人智熄的调试
如题,本人想在开源项目Hyperspectral-Classification Pytorch中加入自定义的网络。
然后产生了一系列的bug,差不多耗费了我一整天出头的时间。
这里把加入自定义网络的流程写出来。同时把调试bug的过程也记录下来。
加入自定义网络
模型读取模块
这部分是在确定模型后,初始化模型相关的超参数,并将模型实例化(即由类创建对象)。
需要在get_model(name, **kwargs):
函数中加入下面的代码。
if name == 'nn':
……
……
elif name ==