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

chess-gpt-eval-contrastive/mamba_module.py

@@ -146,7 +146,7 @@ class MambaPlayer:
                 }
                 wandb.init(project="mamba_linear_probes", name=f"mamba_linear_probes")
                 self.wandb_step = 0
-                self.linear_save_ct += 1
+                self.linear_save_ct = 0
 
     def get_mamba_response(self, game_state: str, temperature: float, max_new_tokens: int, top_k: int):
         game_state = game_state.split("\n\n")[-1].strip()