最近在学习pytorch, 使用mobilenet v2 搭建了一个简单的多标签网络。
全连接层,我替换掉了。两个属性分别使用两个连接层。
本来以为很简单,却一直有问题。
显示如下错误:debug TypeError: dtype must be a type, str, or dtype object
因为是新手,不确定问题在哪里。
首先是怀疑自己写的自定义dataloader有问题。将数据嫁接到因一个神经网络,没有显示以上问题。
然后,一步步替换,神经网络的fc层。
问题一直在,mobilenet v2换成resnet18也不可以。
最后,用了一个之前写的mobilenet v2网络竟然可以了。
比对来比对去,发现除了名字不一样,其余都可以。
试了试,将self.type换成别的名字,竟然可以了。
搞了几个小时,全网搜了一遍,没有找到解法。
主要是不知道问题在哪,通过排除法,终于解决了问题,开心。
神经网络训练终于跑了起来。