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

Change in code after pytorch upgrade

Open CompositeCoding opened this issue 3 years ago • 0 comments

HI,

on line 74: action = torch.argmax(prediction)[0]

should be:

action = torch.argmax(prediction)

after that the code ran

CompositeCoding avatar Jan 21 '22 20:01 CompositeCoding