莫煩pytorch教程中DQN代碼IndexError: too many indices for array

最初代碼: action = torch.max(actions_value, 1)[1].data.numpy()[0, 0] # return the argmaxgit 修改後代碼: action = torch.max(actions_value, 1)[1].data.numpy()[0] # return the argmaxgithub 參考: https://morvanzhou.
相關文章
相關標籤/搜索