print( f"DQN: [{mancala1}] {dqn}\nPlayer: {player} [{mancala2}]") PATH_DQN = "Replay_dqn_state_dict.pt" DQN_model = DQN_replay(14, 6) DQN_model.model.load_state_dict(torch.load(PATH_DQN)) game = Game() done = False while not done: # Player 1 Move board1 = game.board() game._player_one = True p1_action = DQN_model._move(game) print(f"DQN Action: {p1_action + 1}") game.move(p1_action) # End game if move meets win condition if game.over(): break # Player 2 Move render(game.board()) game._player_one = False p2_action = input("Enter an Action between 1 and 6: ") p2_action = 13 - (int(p2_action)) print(p2_action) game.move(p2_action) render(game.board())