Spaces:
Running
Running
Update model_att.py
Browse files- model_att.py +2 -2
model_att.py
CHANGED
|
@@ -88,10 +88,10 @@ class ParallelFeatureExtractorWithAttention(nn.Module):
|
|
| 88 |
return v_cnn, v_bilstm
|
| 89 |
|
| 90 |
|
| 91 |
-
class
|
| 92 |
|
| 93 |
def __init__(self, esm_dim, additional_dim, cnn_out_channels, lstm_hidden_dim, num_classes, dropout_rate=0.42):
|
| 94 |
-
super(
|
| 95 |
|
| 96 |
|
| 97 |
fused_input_dim = esm_dim + additional_dim
|
|
|
|
| 88 |
return v_cnn, v_bilstm
|
| 89 |
|
| 90 |
|
| 91 |
+
class AVP_Fusion(nn.Module):
|
| 92 |
|
| 93 |
def __init__(self, esm_dim, additional_dim, cnn_out_channels, lstm_hidden_dim, num_classes, dropout_rate=0.42):
|
| 94 |
+
super(AVP_Fusion, self).__init__()
|
| 95 |
|
| 96 |
|
| 97 |
fused_input_dim = esm_dim + additional_dim
|