相关疑难解决方法(0)

在PyTorch中调用forward方法与调用模型实例

我看过的很多 PyTorch 教程都是这样做的。

定义模型:

class Network(nn.Module):
    def __init__():
        super().__init__()
        self.conv1 = ..
        ... 
    
    def forward(x)
        ...
    ...
Run Code Online (Sandbox Code Playgroud)

一旦网络被实例化 ( net = Network()),教程中的人就会编写net(input_data)而不是net.forward(input_data). 我尝试过net.forward(),它给出了与 相同的结果net()

为什么这是一种常见做法,以及为什么它有效?

python oop class pytorch

15
推荐指数
1
解决办法
6454
查看次数

标签 统计

class ×1

oop ×1

python ×1

pytorch ×1