diff --git a/engine/src/training/train.rs b/engine/src/training/train.rs index d6b7bd1..a59deec 100644 --- a/engine/src/training/train.rs +++ b/engine/src/training/train.rs @@ -88,6 +88,8 @@ pub fn train(training_config: TrainingConfig, device: B::Dev let mut board_state = BoardState::default(); let mut episode_buffer: Vec = vec![]; + let mut game_hist = String::new(); + while board_state.status == BoardStateStatus::Ongoing { let results = mcts.search(&board_state, &infer_model, &device); episode_buffer.push(results); @@ -102,9 +104,15 @@ pub fn train(training_config: TrainingConfig, device: B::Dev let stm = board_state.board.side_to_move(); let mv = sample_move(&adjusted, &mut rng, stm).unwrap(); - println!("playing move: {}", mv); + // println!("playing move: {}", mv); + if episode == 0 { + game_hist.push_str(mv.to_string().as_str()); + game_hist.push(' '); + } + board_state.apply_move(mv) } + println!("Game history of first game of iteration: {}", game_hist); for result in episode_buffer.iter().enumerate() { if board_state.status == BoardStateStatus::Stalemate