Update chess-gpt-eval-contrastive/mamba_module.py

chess-gpt-eval-contrastive/mamba_module.py

@@ -316,7 +316,7 @@ class MambaPlayer:
                             self.linear_probes[layer_idx][probe_type].fit(X, y)
                             print(f"Fit layer {layer_idx} type {probe_type}.")
                         else:
-                            print(f"Skipping training for layer {layer_idx}, bucket {bucket}, probe type {probe_type} due to inconsistent number of samples.")
+                            print(f"Skipping training for layer {layer_idx}, bucket {bucket}, probe type {probe_type} due to inconsistent number of samples. X shaspe {X.shape}, Y shape {y.shape}")
     
         # Reset linear_probe_targets after training
         self.linear_probe_targets = {i: {bucket: {'q_value': [], 'q_value_delta': [], 'material_balance': []} for bucket in self.move_buckets} for i in self.linear_probes}