深度学习模型中,像forward()传参,明明已经传入了正确数量的参数,却报错TypeError: forward() takes 2 positional arguments but 3 were given。
查阅资料发现,使用nn.Sequential()定义的网络,只能接受单输入,同时输出也只能是单输出。
解决方法,可以将两个tensor经过torch.cat组合在一起,然后在forward()中将组合拆分。
深度学习模型中,像forward()传参,明明已经传入了正确数量的参数,却报错TypeError: forward() takes 2 positional arguments but 3 were given。
查阅资料发现,使用nn.Sequential()定义的网络,只能接受单输入,同时输出也只能是单输出。
解决方法,可以将两个tensor经过torch.cat组合在一起,然后在forward()中将组合拆分。