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

Update train.py

Open bigrobinson opened this issue 5 years ago • 0 comments

type: feat

Fully soft-code optimizer choice between Adam and SGD as well as learning rate. Remove [0] index on line 79 to prevent IndexError: invalid index of a 0-dim tensor.

bigrobinson avatar Dec 09 '19 20:12 bigrobinson