Update chess-gpt-eval/main.py

chess-gpt-eval/main.py

@@ -611,6 +611,12 @@ def play_games(
     if isinstance(player_two, StockfishPlayer):
         player_two.close()
 
+    if RUN_FOR_ANALYSIS:
+        csv_file_path = f"logs/{player_one_recording_name}_vs_{player_two_recording_name}"
+        csv_file_path = csv_file_path.replace(".", "_") # Because I'm using ckpt filenames for nanogpt models 
+        csv_file_path += ".csv"
+    else:
+        csv_file_path = recording_file
     stats = calculate_stats(csv_file_path)
     if stats:
         print("\nStatistics:")
@@ -636,7 +642,7 @@ random_opening_moves = 20
 if __name__ == "__main__":
     for nanogpt_player in player_ones:
         for i in [0]: # [3] #range(11):
-            num_games = 545
+            num_games = 500
             # player_one = GPTPlayer(model="gpt-3.5-turbo-instruct")
             # player_one = LocalLlamaPlayer(model_name="meta-llama/Llama-2-7b-hf")
             # player_one = LocalLoraLlamaPlayer("meta-llama/Llama-2-7b-hf", "/workspace/axolotl/lora2-out")