Получение ошибки "AttributeError: объект 'NoneType' не имеет атрибута 'shape'" при реализации Atari Breakout

Question

Я сделал код для решения Atari Breakout. У меня небольшая проблема, но я не могу сказать, в чем она.

Вот код

Это проблема с памятью воспроизведения.

try:
    next_states = torch.tensor(batch[3], dtype=torch.float32) 
except:
    import ipdb; ipdb.set_trace()

Проблема в том, где эти строки. set_trace()используется для вызова интерактивной оболочки. Оттуда, если я сбегуfor i in range(batch_size): print(batch[3][i].shape), Я получил этот вывод

ipdb> for i in range(32): print(batch[3][i].shape)
(4, 84, 84)
(4, 84, 84)
(4, 84, 84)
(4, 84, 84)
(4, 84, 84)
(4, 84, 84)
(4, 84, 84)
(4, 84, 84)
(4, 84, 84)
(4, 84, 84)
(4, 84, 84)
(4, 84, 84)
(4, 84, 84)
(4, 84, 84)
(4, 84, 84)
(4, 84, 84)
(4, 84, 84)
(4, 84, 84)
(4, 84, 84)
(4, 84, 84)
(4, 84, 84)
(4, 84, 84)
(4, 84, 84)
(4, 84, 84)
(4, 84, 84)
(4, 84, 84)
(4, 84, 84)
*** AttributeError: 'NoneType' object has no attribute 'shape'

Как я могу улучшить этот код, чтобы избежать такой ошибки?

Answer 1

Ошибка говорит вам о проблеме. Ты пытаешься позвонитьshape на None, поэтому в вашем коде некоторая переменная a является None и ты звонишь shape на нем, т.е. a.shape. Это одна из самых распространенных ошибок в программировании!

В вашем for петля

for i in range(32): 
    print(batch[3][i].shape)

в какой-то момент, batch[3][i] является None, так что вам придется выяснить, что batch[3] содержит и почему это None.

См. Обсуждение здесь https://chat.stackexchange.com/transcript/message/54070403.