RuntimeError: invalid argument 2: size '[-1 x 4032]'

[richielo]

Hello,

I am facing an issue of even running the sample training command

python3 pytorch_rl/main.py --no-vis --env-name Duckietown-small_loop-v0 --algo a2c --lr 0.0002 --max-grad-norm 0.5 --num-steps 20

The error message happens on line 116 in model.py

x = x.view(-1, 32 * 9 * 14)

Error message:

RuntimeError: invalid argument 2: size '[-1 x 4032]' is invalid for input with 127872 elements at ..\aten\src\TH\THStorage.cpp:80

I am using windows and installed everything with the first provided method using pip. I have not changed anything in the code, looks like size mismatch with the convnet. Thanks for your help in advance

[fenjiro]

hello, i had the same problem, after installing gym-duckietown in ubutu 16.06 environement using conda, have you got any feedback or tips to resolve the problem ?

File "/home/osboxes/gym-duckietown/pytorch_rl/model.py", line 115, in forward x = x.view(-1, 32 9 14) RuntimeError: invalid argument 2: size '[-1 x 4032]' is invalid for input with 127872 elements at /opt/conda/conda-bld/pytorch_1523244252089/work/torch/lib/TH/THStorage.c:37

[Guangyi-Z]

Same here on anaconda3-5.2.0 on OSX.

[bhairavmehta95]

Hi everyone. Thank you all for reporting, and sorry we couldn't see this sooner.

We wrote the pytorch RL code for this much earlier in the development process, and haven't gotten around to updating it with the new changes. This has to do with the size of the observation (we wrote it for 160x120, but now the default size is 640x480). Changing that line (the flatten one) will fix the issue, and we'd love for one of you to help us with a PR if you were able to solve the issue.

For the AIDO, you can take a look here at a more updated version of the RL code: https://github.com/duckietown/challenge-aido1_LF1-baseline-RL-sim-pytorch