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

chess-gpt-eval-contrastive/main.py

@@ -625,30 +625,22 @@ def play_games(
                 illegal_move_numbers
             )
             games_saved += 1
-        else:
-            print("Duplicate game; not saved.")
 
-        if update_activations:
-            if player_one_resignation or player_one_failed_to_find_legal_move:
-                player_one.update_activations("lost")
-                #player_two.update_activations("won")
-            elif player_two_resignation or player_two_failed_to_find_legal_move:
-                player_one.update_activations("won")
-                #player_two.update_activations("lost")
-            else:
-                if board.result() == "1-0":
-                    player_one.update_activations("won")
-                 #   player_two.update_activations("lost")
-                elif board.result() == "0-1":
+            if update_activations:
+                if player_one_resignation or player_one_failed_to_find_legal_move:
                     player_one.update_activations("lost")
-                  #  player_two.update_activations("won")
-                #else:  # Draw
-                 #   player_one.update_activations("draw")
-                   # player_two.update_activations("draw")
+                elif player_two_resignation or player_two_failed_to_find_legal_move:
+                    player_one.update_activations("won")
+                else:
+                    if board.result() == "1-0":
+                        player_one.update_activations("won")
+                    elif board.result() == "0-1":
+                        player_one.update_activations("lost")
         
-            if games_saved % save_activations_every == 0:
-                player_one.save_activations(activations_path)
-            #player_two.save_activations(activations_path)
+                if games_saved % save_activations_every == 0:
+                    player_one.save_activations(activations_path)
+        else:
+            print("Duplicate game; not saved.")
             
     if isinstance(player_one, StockfishPlayer):
         player_one.close()
@@ -689,7 +681,7 @@ save_activations_every = 25
 if __name__ == "__main__":
     for nanogpt_player in player_ones:
         for i in [0]: # [3] #range(11):
-            num_games = 2500
+            num_games = 2500-65
             # 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")