yezdata
/

EmCoder

Text Classification

emotion-recognition

bayesian-deep-learning

uncertainty-quantification

multi-label-classification

Eval Results (legacy)

Model card Files Files and versions

yezdata commited on May 18

Commit

54ed2c9

·

verified ·

1 Parent(s): 2625b05

Update modeling_emcoder.py

Files changed (1) hide show

modeling_emcoder.py +1 -1

modeling_emcoder.py CHANGED Viewed

@@ -25,7 +25,7 @@ class EmCoderEncoder(nn.Module):
             batch_first=True,
         )
         self.encoder = nn.TransformerEncoder(
-            encoder_layer=encoder_layer, num_layers=config.n_layers
         )
         self.final_norm = nn.LayerNorm(config.d_model)

             batch_first=True,
         )
         self.encoder = nn.TransformerEncoder(
+            encoder_layer=encoder_layer, num_layers=config.n_layers, enable_nested_tensor=False
         )
         self.final_norm = nn.LayerNorm(config.d_model)