-
Notifications
You must be signed in to change notification settings - Fork 124
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Error Running Train.py #6
Comments
I meet the question so. It cause bytorch version. In latest torch, 0-dim tensor is invaild, Change code line77
To
|
I tried that. Still errors (python36) C:\Flappy-bird-deep-Q-learning-pytorch>python train.py |
@2448845600 thanks, guy! fix my problem. |
@2448845600 谢谢~搞定啦 |
Flappy-bird-deep-Q-learning-pytorch/src/deep_q_network.py:21: UserWarning: nn.init.uniform is now deprecated in favor of nn.init.uniform_.
nn.init.uniform(m.weight, -0.01, 0.01)
Perform a random action
Iteration: 2/2000000, Action: 1, Loss: 0.010123813524842262, Epsilon 0.1, Reward: 0.1, Q-value: 0.0005683371564373374
Traceback (most recent call last):
File "train.py", line 133, in
train(opt)
File "train.py", line 74, in train
action = torch.argmax(prediction)[0]
IndexError: invalid index of a 0-dim tensor. Use tensor.item() to convert a 0-dim tensor to a Python number
(python36) bash-3.2$
The text was updated successfully, but these errors were encountered: