sheeprl
sheeprl copied to clipboard
TransDreamer
I don't know if you have ever considered but TSSM idea doesn't look bad and it can be a good add-on for this library.
Official Paper: https://arxiv.org/pdf/2202.09481
Implementation: https://github.com/changchencc/TransDreamer
I don't know why they compared it against the dreamer-v2 only.
That would awesome. Have you considered to open a PR with an initial draft?
Maybe in the near future but at this very moment I am quite busy with my on going work. I will visit your PRs and issues, if there is going to be no PR related to this then I will start working on it.