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

chess-gpt-eval-contrastive/main.py

@@ -531,15 +531,15 @@ def play_games(
                 f.write(game_state)
             current_move_num = f"{board.fullmove_number if move_num_in_gamestate else ''}."
 
-            if total_moves == random_move_start:
-                for _ in range(10):
-                    g, b, opening_moves = add_random_moves(game_state, board, random_opening_moves)
-                    if g is not None:
-                        game_state = g
-                        board = b
-                        break
-                total_moves += random_opening_moves
-                continue
+            #if total_moves == random_move_start:
+            #    for _ in range(10):
+            #        g, b, opening_moves = add_random_moves(game_state, board, random_opening_moves)
+            #        if g is not None:
+            #            game_state = g
+            #            board = b
+            #            break
+            #    total_moves += random_opening_moves
+            #    continue
 
 
             total_moves += 1
@@ -679,7 +679,7 @@ player_ones = ["50M/anneal/anneal_complete_round3.pt"]
 player_two_recording_name = "lc0_sweep" #"stockfish_sweep"
 move_num_in_gamestate = False
 book_opening = True
-random_opening = True
+random_opening = False
 random_opening_moves = 10
 contrastive_activation_save_interval = 10
 activations_path="activations.pkl"
@@ -698,7 +698,7 @@ if __name__ == "__main__":
 #            player_one = NanoGptPlayer(model_name=player_one_recording_name, move_num_in_gamestate=move_num_in_gamestate)
             #player_one_recording_name = f"xformer_rdm_{rm}"
             player_one = MambaPlayer(model_name=player_one_recording_name, move_num_in_gamestate=move_num_in_gamestate)
-            player_one_recording_name = f"random_mamba_start/mamba_rdmstart_{rm}"
+            player_one_recording_name = f"mamba_contrastive"
             if not update_activations:
                 player_one.apply_contrastive_activations()