HaileyStorm commited on
Commit
5994510
1 Parent(s): a7153f4

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

Browse files
Files changed (1) hide show
  1. chess-gpt-eval-contrastive/main.py +1 -1
chess-gpt-eval-contrastive/main.py CHANGED
@@ -767,7 +767,7 @@ if __name__ == "__main__":
767
  # player_one = NanoGptPlayer(model_name=player_one_recording_name, move_num_in_gamestate=move_num_in_gamestate)
768
  #player_one_recording_name = f"xformer_rdm_{rm}"
769
  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)
770
- player_one_recording_name = f'linear_train'
771
  if apply_activations:
772
  player_one.apply_contrastive_activations(path=activations_path, weight=wgt)
773
 
 
767
  # player_one = NanoGptPlayer(model_name=player_one_recording_name, move_num_in_gamestate=move_num_in_gamestate)
768
  #player_one_recording_name = f"xformer_rdm_{rm}"
769
  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)
770
+ player_one_recording_name = f'linear_eval'
771
  if apply_activations:
772
  player_one.apply_contrastive_activations(path=activations_path, weight=wgt)
773