描述:用Pytorch写模型的时候,报错:list object is not callable
原因:
很多模型中间会有几层是完全相同的,就有人用[nn.Conv2D() for _ in range(num)]之类的一行简单语句实现几层的堆叠。这样定义的网络是一个列表,需要用nn.Sequential()包装一下才能在forward函数里实现数据传输。没用nn.Sequential()包装时会报错。
解决方法如上。
描述:用Pytorch写模型的时候,报错:list object is not callable
原因:
很多模型中间会有几层是完全相同的,就有人用[nn.Conv2D() for _ in range(num)]之类的一行简单语句实现几层的堆叠。这样定义的网络是一个列表,需要用nn.Sequential()包装一下才能在forward函数里实现数据传输。没用nn.Sequential()包装时会报错。
解决方法如上。