AssertionError: action shape did not match expected=(1,) actual=(1, 1)

1

Hello,

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.

2

I guess you might be using different PyTorch versions, where this additional check was either added or removed.
Could you post the stack trace so that we can have a look at the error?