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

chess-gpt-eval-contrastive/main.py

@@ -565,6 +565,8 @@ def play_games(
             if illegal_moves_one != 0:
                 player_one_legal_moves -= 1
                 illegal_move_numbers.append(board.fullmove_number)
+            if update_activations:
+                player_one.update_activations("current")
             if (
                 board.is_game_over()
                 or player_one_resignation
@@ -644,7 +646,6 @@ def play_games(
                  #   player_one.update_activations("draw")
                    # player_two.update_activations("draw")
         
-        if update_activations and games_saved % contrastive_activation_save_interval == 0:
             player_one.save_activations(activations_path)
             #player_two.save_activations(activations_path)