Flappy-bird-deep-Q-learning-pytorch icon indicating copy to clipboard operation
Flappy-bird-deep-Q-learning-pytorch copied to clipboard

Published 20 hours ago • verified publisher icon uvipen

Error in dimension of image tensor

Open bigrobinson opened this issue 5 years ago • 0 comments

First of all, thanks for taking the time to put this pytorch implementation together. I am getting the following error. Also can you explain what the purpose of the torch.cat line is?

Traceback (most recent call last): File "train.py", line 133, in train(opt) File "train.py", line 58, in train state = torch.cat(tuple(image for _ in range(4)))[None, :, :, :] IndexError: too many indices for tensor of dimension 2

bigrobinson avatar Dec 05 '19 18:12 bigrobinson