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

chess-gpt-eval-contrastive/main.py

@@ -766,7 +766,7 @@ if __name__ == "__main__":
             player_one_recording_name = nanogpt_player
 #            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, update_contrastive=update_activations, update_linear=update_linear, linear_probe_path=linear_path)
+            player_one = MambaPlayer(model_name=player_one_recording_name, move_num_in_gamestate=move_num_in_gamestate, update_contrastive=update_activations, update_linear=update_linear or eval_linear, linear_probe_path=linear_path)
             player_one_recording_name = f'linear_eval'
             if apply_activations:
                 player_one.apply_contrastive_activations(path=activations_path, weight=wgt)