I am trying to replicate the following code: https://github.com/ikostrikov/pytorch-a2c-ppo-acktr-gail and run it on a server. However, when I run it on the server side, I get the following error:
AssertionError: action shape did not match expected=(1,) actual=(1, 1)
It seems like there is a problem with the dimensions of the action space, but I don’t get the same Error when I try to train it locally on my laptop (only CPU).
Thank you in advance.