大佬，你好，我在用pytorch训练出来的模型，无法加载运行，报了如下错误

[ToBeIronMan]

Traceback (most recent call last): File "human_play.py", line 88, in run() File "human_play.py", line 81, in run game.start_play(human, mcts_player, start_player=0, is_shown=1) File "C:\Users\lucky\Desktop\AlphaZero_Gomoku-master\AlphaZero_Gomoku-master\game.py", line 177, in start_play move = player_in_turn.get_action(self.board) File "C:\Users\lucky\Desktop\AlphaZero_Gomoku-master\AlphaZero_Gomoku-master\mcts_alphaZero.py", line 190, in get_action acts, probs = self.mcts.get_move_probs(board, temp) File "C:\Users\lucky\Desktop\AlphaZero_Gomoku-master\AlphaZero_Gomoku-master\mcts_alphaZero.py", line 147, in get_move_probs self._playout(state_copy) File "C:\Users\lucky\Desktop\AlphaZero_Gomoku-master\AlphaZero_Gomoku-master\mcts_alphaZero.py", line 122, in _playout action_probs, leaf_value = self._policy(state) File "C:\Users\lucky\Desktop\AlphaZero_Gomoku-master\AlphaZero_Gomoku-master\policy_value_net_numpy.py", line 104, in policy_value_fn X = relu(conv_forward(X, self.params[i], self.params[i+1])) TypeError: 'int' object is not subscriptable

[syys96]

pytorch训练的模型直接用pytoch加载不更好吗

[yaohaizhou]

应该在human_play.py中把 best_policy = PolicyValueNetNumpy(width, height, policy_param)改为 best_policy = PolicyValueNet(width, height, policy_param) 同时最上面解除 from policy_value_net_pytorch import PolicyValueNet # Pytorch 的注释

[yaohaizhou]

还要把human_play.py中的 best_policy = PolicyValueNet(width, height, policy_param) 改为 best_policy = PolicyValueNet(width, height, model_file) 也就是说不需要使用pickle