Closed ghost closed 3 years ago
Hi, I test it on my machine with torch==1.1.0
and torch==1.7.1
, but I can't reproduce this error. Could I get the version of the pytorch on your machine, please? And one suggestion to solve this problem is to replace .view()
with .reshape()
, Could you try it firstly?
python3.6, torch==1.7.0+cu101 on google colab, please find the gist here
@emadboctorx Thanks for your information, and I managed to reproduce the same error on google colab. The solution is to replace .view()
with .reshape()
in https://github.com/TianhongDai/reinforcement-learning-algorithms/blob/master/rl_algorithms/a2c/models.py#L28 and I can run it on colab now.
Thanks, I'll modify the code accordingly
Hi, after I installed using
pip install -e .
, I tried to run:in
reinforcement-learning-algorithms/rl_algorithms/a2c
and I get the following error on google colab: