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

chess-gpt-eval-contrastive/main.py

@@ -731,7 +731,7 @@ def play_games(
 
 
 RUN_FOR_ANALYSIS = True
-MAX_MOVES = 999 # Due to nanogpt max input length of 1024
+MAX_MOVES = 60 #999 # Due to nanogpt max input length of 1024
 recording_file = "logs/determine.csv" # default recording file. Because we are using list [player_ones], recording_file is overwritten
 player_ones = ["50M/anneal/anneal_complete_round3.pt"]
 player_two_recording_name = "lc0_sweep" #"stockfish_sweep"
@@ -747,8 +747,8 @@ save_activations_every = 25
 contrastive_weight = 0.8
 
 linear_path="linear.pkl"
-update_linear = True
-eval_linear = False
+update_linear = False
+eval_linear = True
 if __name__ == "__main__":
     for nanogpt_player in player_ones:
         i = 0