Change in code after pytorch upgrade
CompositeCoding opened this issue · comments
HI,
on line 74:
action = torch.argmax(prediction)[0]
should be:
action = torch.argmax(prediction)
after that the code ran
uvipen / Flappy-bird-deep-Q-learning-pytorch
Deep Q-learning for playing flappy bird game