Closed LongchaoDa closed 2 years ago
LSTM is not supported in the pretraining. There was a PR adding this but it has then died out: https://github.com/hill-a/stable-baselines/pull/315 . For better imitation learning algos see imitation library, for example. Note that SB2 is not mantained anymore.
You may close this issue if you have no further questions.
Thank you, so you mean the structure of Stable Baseline 2 is difficult to support custom policy LSTM into GAIL traning? Maybe i will turn to the link you mentioned to explore more info.
Thank you, so you mean the structure of Stable Baseline 2 is difficult to support custom policy LSTM into GAIL traning? Yes it requires a bit of work (you may check the PR #315).
Closing issue as resolved :).
I was training a GAIL model with MlpLstmPolicy in Stable_Baselines2, however, I could not successfully run the training process even though: I made the
in the TRPO part, Is there any changes I should make? Or If there is no other possible solution, how can i customize a LSTM policy to be compatible with GAIL for myself?
Looking forward to your reply!
The error happened is here: