Update s2d CrossEncoder.py

chungimungi · Oct 16, 2023 · a6ee6e0 · a6ee6e0
1 parent 29b9c14
commit a6ee6e0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/s2d CrossEncoder.py b/s2d CrossEncoder.py
@@ -49,7 +49,7 @@ def __getitem__(self, index):
 
         return torch.LongTensor(padded_input_seq), torch.LongTensor([label])
 
-# Define your custom model architecture
+# model architecture
 class CustomCrossEncoder(nn.Module):
     def __init__(self, vocab_size, embed_dim, hidden_dim, num_classes, dropout_prob=0.5):
         super(CustomCrossEncoder, self).__init__()