typeError: bark() takes 1 positional argument but 2 were given

Question

前提・実現したいこと

typeError: bark() takes 1 positional argument but 2 were givenというエラーがでました。
文字列が二つの位置引数と認識されているみたいですが、
どうやって解決したらいいでしょうか

line 95:net = FujiNet(num_classes).to(device)

発生している問題・エラーメッセージ

$ python FujiNet.py 
Traceback (most recent call last):                                         
  File "FujiNet.py", line 95, in <module>
    net = FujiNet(num_classes).to(device)
TypeError: __init__() takes 1 positional argument but 2 were given

該当のソースコード

class FujiNet(nn.Module):
    def __init__(self):
        super(FujiNet, self).__init__()
        self.pool = nn.MaxPool3d(2, 2)
        self.conv1 = nn.Conv3d(1, 3, 3, padding=1)
        self.conv2 = nn.Conv3d(3, 3, 3, padding=1)
        self.conv3 = nn.Conv3d(3, 32, 3, padding=1)
        self.conv4 = nn.Conv3d(32, 64, 3, padding=1)
        self.fc1 = nn.Linear(10 * 10 * 10 * 64, 512)
        self.fc2 = nn.Linear(512, 4)
        self.dropout = nn.Dropout(p=0.5)

    def forward(self, x):
        x = self.pool(x)
        x = F.relu(self.conv1(x))
        x = F.relu(self.conv2(x))
        x = self.pool(x)
        x = F.relu(self.conv3(x))
        x = F.relu(self.conv4(x))
        x = self.pool(x)
        x = x.view(-1, 10 * 10 * 10 * 64)
        x = self.dropout(x)
        x = F.relu(self.fc1(x))
        x = self.dropout(x)
        x = self.fc2(x)
        return x

# select device
num_classes = 5
device = 'cuda' if torch.cuda.is_available() else 'cpu'
net = FujiNet(num_classes).to(device)

Answer 1

class FujiNet(nn.Module) の中身を見ますと、def __init__(self):となっていますので、FujiNet(num_classes)のように引数を取ることはできないと思います。インスタンスの初期化をする際に num_classes を使用する部分は無いようにお見受けします。引数を使用するのは forward の方ではないでしょうか？
以下のコードで実現できそうです。

# インスタンス初期化
fjn = FujiNet()
# forwardに引数 num_classes を渡す
net = fjn.forward(num_classes).to(device)

Comments

[Hiro051]

解決できました。
ありがとうございます！