TVconv与MobilenetV2
将TVconv替换DWconv后,根据源代码中的参数。
shape ‘[32, 8, 9, 96, 96]’ is invalid for input of size 28901376
这个错误
原因是view后,对不上self.unfold(x),需要更改h_w的值。让乘积等于28901376。将所有的h_w乘以1.16666,然后四舍五入即可。
TVconv与MobilenetV2
将TVconv替换DWconv后,根据源代码中的参数。
shape ‘[32, 8, 9, 96, 96]’ is invalid for input of size 28901376
这个错误
原因是view后,对不上self.unfold(x),需要更改h_w的值。让乘积等于28901376。将所有的h_w乘以1.16666,然后四舍五入即可。