PyTorchで、initにシーケンシャルで書く方法と、forwardに畳み込みなどの処理を書く方法とがありますが、どちらかが、処理が速いのですか?
Python
1# シーケンシャルな書き方の例 2class Net(nn.Module): 3 def __init__(self): 4 super(Net, self).__init__() 5 conv = [ nn.Conv2d(3, 128, 3), 6 nn.BatchNorm2d(128), 7 nn.ReLU(), 8 nn.Conv2d(128, 128, 3), 9 nn.BatchNorm2d(128)] 10 11 self.conv = nn.Sequential(*conv) 12 13 def forward(self, x): 14 return self.conv(x)
バッドをするには、ログインかつ
こちらの条件を満たす必要があります。