Усиление обучения с Pytorch. [Ошибка: KeyError] - PullRequest
Новая
       21

Усиление обучения с Pytorch. [Ошибка: KeyError]

0 голосов
/ 08 июля 2019

Я новичок в подкреплении обучения, а также pytorch.Я учу это у Удеми.Тем не менее, код у меня такой же, как показано, но у меня ошибка.Я думаю, что это ошибка Pytorch, но не могу отладить ее.Если кто-то помогает, это действительно ценится.

import gym
import time
import torch
import matplotlib.pyplot as plt
from gym.envs.registration import register
register(id='FrozenLakeNotSlippery-v0',entry_point='gym.envs.toy_text:FrozenLakeEnv',kwargs={'map_name': '4x4', 'is_slippery':False})
env = gym.make('FrozenLakeNotSlippery-v0')
number_of_states = env.observation_space.n
number_of_actions = env.action_space.n
Q = torch.zeros([number_of_states,number_of_actions])
num_episodes = 1000
steps_total = []
gamma = 1
for i in range(num_episodes):
    state = env.reset()
    step = 0
    while True:
        step += 1
        #action = env.action_space.sample()
        random_values = Q[state]+torch.rand(1,number_of_actions)/1000
        action = torch.max(random_values,1)[1][0]
        new_state, reward, done, info = env.step(action)
        Q[state, action] = reward + gamma * torch.max(Q[new_state])
        state = new_state
        #time.sleep(0.4)
        #env.render()
        if done:
            steps_total.append(step)
            print ("Episode Finished after %i steps" %step)
            break

print ("Average Num Steps: %2f" %(sum(steps_total)/num_episodes))
plt.plot(steps_total)
plt.show()

У меня ошибка, указанная ниже 

KeyError                                  Traceback (most recent call last)
<ipython-input-11-a6aa419c3767> in <module>
  8         random_values = Q[state]+torch.rand(1,number_of_actions)/1000
  9         action = torch.max(random_values,1)[1][0]
---> 10         new_state, reward, done, info = env.step(action)
 11         Q[state, action] = reward + gamma * torch.max(Q[new_state])
 12         state = new_state

c:\users\souradip\appdata\local\programs\python\python36\lib\site-packages\gym\envs\toy_text\discrete.py in step(self, a)
 53 
 54     def step(self, a):
---> 55         transitions = self.P[self.s][a]
 56         i = categorical_sample([t[0] for t in transitions], self.np_random)
 57         p, s, r, d= transitions[i]

KeyError: tensor(3)

1 Ответ

1 голос
/ 08 июля 2019

код ниже 

action = torch.max(random_values,1)[1][0]

дает тензор 0-dim, но env.step() ожидает число питона, которое в основном является действием из пространства действий. Так, как @ a_guest , упомянутый в комментарии, используйте a.item(), чтобы преобразовать тензор 0-dim в число питона, как показано ниже: 

new_state, reward, done, info = env.step(action.item())
...