Pytorch/複数入力の深層学習モデルの構築がうまくできません

Question

pytorch初心者です。
CNNの複数入力モデルを構築したいのですが、関数内の定義がうまくいかず
local variable 'x_bs' referenced before assignmentのエラーが出ます。

入力は画像とほかmetadataが2つ（1次元、26次元）の計3変数、目的変数は連続データ、ResNet50のファインチューニング回帰モデルを目指しています。
コードは少し長くなり恐縮ですが、datasetから説明させていただきます。

#dataloader作成
df_train = torch.utils.data.TensorDataset(x_train,x_bs_train, x_cate_train, y_train)
df_val = torch.utils.data.TensorDataset(x_val,x_bs_val, x_cate_val, y_val)

batch_size=32

train_dataloader = torch.utils.data.DataLoader(df_train, batch_size=batch_size, shuffle=True)
val_dataloader = torch.utils.data.DataLoader(df_val, batch_size=batch_size, shuffle=False)

#モデル構築
class MyModel(nn.Module):
    def __init__(self):
        super(MyModel, self).__init__()
        # 画像
        self.resnet = nn.Sequential(*list(models.resnet50(pretrained=True).children())[:-1])
        self.image_fc = nn.Linear(2048, 256)
        # metadata：bs1次元,cate26次元
        self.bs_fc = nn.Linear(1, 256)
        self.cate_fc = nn.Linear(26, 256)
        self.all_fc = nn.Linear(256*3, 256*3) 
        self.dropout = nn.Dropout(0.25)
        # 最終FC層
        self.last_fc = nn.Linear(256, 1)
        
    def forward(self, image, bs, cate):
        # 画像features
        x_img = self.resnet(image).view(-1, 2048) 
        x_img = F.relu(self.image_fc(x_img))
        # metadata_features
        x_bs = F.relu(self.bs_fc(x_bs))
        x_cate = F.relu(self.bs_fc(x_cate))
        # concat,relu, dropout
        x = torch.cat([x_img, x_bs, x_cate], 1)
        x = F.relu(self.all_fc(x))
        x = self.dropout(x)
        # 最終FC
        y = self.last_fc(x)
        return y
    
model = MyModel()

#モデルが挙動するか確認
for x_train,x_bs_train, x_cate_train, y_train in train_dataloader:
    y_pred = model(x_train,x_bs_train,x_cate_train)

#エラー：
UnboundLocalError                         Traceback (most recent call last)
<ipython-input-15-181704a40564> in <module>
      1 for x_train,x_bs_train, x_cate_train, y_train in train_dataloader:
----> 2     y_pred = model(x_train,x_bs_train,x_cate_train)

~\AppData\Local\Continuum\anaconda3\lib\site-packages\torch\nn\modules\module.py in __call__(self, *input, **kwargs)
    530             result = self._slow_forward(*input, **kwargs)
    531         else:
--> 532             result = self.forward(*input, **kwargs)
    533         for hook in self._forward_hooks.values():
    534             hook_result = hook(self, input, result)

<ipython-input-13-3a3ca0de36f3> in forward(self, image, bs, cate)
     20         x_img = F.relu(self.image_fc(x_img))
     21         # メタデータ特徴
---> 22         x_bs = F.relu(self.bs_fc(x_bs))
     23         x_cate = F.relu(self.bs_fc(x_cate))
     24         # concat,relu, dropout

UnboundLocalError: local variable 'x_bs' referenced before assignment

グローバル変数とローカル変数の不整合にあると思うのですが修正方法がわかりません。

ご教授のほど、よろしくお願いします。

Answer 1

単純な記載ミスでした。

def forward(self, image, bs, cate):
の部分を
def forward(self, image, x_bs, x_cate):
に変える必要がありますね。