Upload model.py

model.py

@@ -66,7 +66,7 @@ class ML_BART(nn.Module):
         super().__init__()
         d_model = bartconfig.d_model
 
-        self.decoder_emb = nn.ModuleList([
+        self.decoder_emb2 = nn.ModuleList([
             nn.Embedding(class_num[0] + 1, d_model // 4),
             nn.Embedding(class_num[1] + 1, d_model // 4)
         ])
@@ -91,7 +91,7 @@ class ML_BART(nn.Module):
             emb_decoder = self.encoder(x_decoder)
         else:
             emb_decoder = torch.concatenate(
-                [self.decoder_emb[0](x_decoder[..., 0]), self.decoder_emb[1](x_decoder[..., 1]),
+                [self.decoder_emb2[0](x_decoder[..., 0]), self.decoder_emb2[1](x_decoder[..., 1]),
                  self.decoder(x_encoder)], dim=-1)
 
         y = self.bart(inputs_embeds=emb_encoder, decoder_inputs_embeds=emb_decoder,