hill-a / stable-baselines

A fork of OpenAI Baselines, implementations of reinforcement learning algorithms
http://stable-baselines.readthedocs.io/
MIT License
4.16k stars 725 forks source link

[Question] Error because of dimension of observation in GAIL #1074

Closed SiweiJu closed 3 years ago

SiweiJu commented 3 years ago

I am trying to implement GAIL with my customized environment. The env check has passed. The dimensions are as followed. image

But later in in model.learn there is an error image

which seems to be due to the reshape in runners.py image