get batch_size and seq_len from encoder outputs instead of inputs

Author: ianbulovic

src/cnlpt/modeling/models/projection_model.py

@@ -293,7 +293,7 @@ def forward(
 
         outputs = self.encoder(input_ids, **kwargs)
 
-        batch_size, seq_len = input_ids.shape
+        batch_size, seq_len, _ = outputs.last_hidden_state.shape
 
         logits = []