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

chess-gpt-eval-contrastive/main.py

@@ -676,7 +676,8 @@ book_opening = False
 random_opening = True
 random_opening_moves = 10
 activations_path="activations_rdm.pkl"
-update_activations = False # False = use them
+update_activations = True
+apply_activations = False
 save_activations_every = 25
 contrastive_weight = 0.8
 if __name__ == "__main__":
@@ -698,7 +699,7 @@ if __name__ == "__main__":
             #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'contrastive_weights_rdm/mamba_rdm_wgt_{wgt}' #f'contrastive_rdm/mamba_rdm_{rm}'
-            if not update_activations:
+            if apply_activations:
                 player_one.apply_contrastive_activations(path=activations_path, weight=wgt)
 
             #player_two = StockfishPlayer(skill_level=i, play_time=0.1)