Wwwy1031 commited on
Commit
21c24ec
·
verified ·
1 Parent(s): bfb576f

Update model_att.py

Browse files
Files changed (1) hide show
  1. 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 AVP_HNCL_v3(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_HNCL_v3, self).__init__()
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