forward一般在模型中都可以看到,然后实际代码中总是看不到它被调用,然后我就尝试着调试了一下代码,发现在
output = model(inputs)
这段代码里面是调用了其中的forward方法的。
forward的作用主要是首先网络各个层的拼接,保证其能够按照一定的顺序来进行。
同时可以在定义子网络时,使用nn.Sequential容器也可以保证网络的流动顺序
总的代码可以参考:
导入训练好的模型,Pytorch
Pytorch 构建神经网络时的forward
最新推荐文章于 2024-04-25 21:44:04 发布