Closed ndeshp2s closed 5 years ago
I tried testing the trained model on A3C on CartPole-v1 environment. However, I get the following error: "ValueError: Error when checking input: expected input_1 to have 3 dimensions, but got array with shape (4, 4)"
Hi @ndeshp2s, try changing the code in A3C/agent.py as follows :
A3C/agent.py
# line 25, in reshape() return np.expand_dims(x, axis=0)
It should work fine for CartPole-v1 after that ;)
I tried testing the trained model on A3C on CartPole-v1 environment. However, I get the following error: "ValueError: Error when checking input: expected input_1 to have 3 dimensions, but got array with shape (4, 4)"