search

kzl / decision-transformer

Official codebase for Decision Transformer: Reinforcement Learning via Sequence Modeling.
MIT License
2.33k stars 440 forks source link

[IDEA] Code for dataset generation #54

Open JustinS6626 opened 1 year ago

JustinS6626 commented 1 year ago

I was wondering if it would be possible to release the code that was used to generate the datasets for the gym and atari experiments. That would facilitate the evaluation of decision transformer methods on other environments.

jbloomAus commented 1 year ago

This would be really good. Furthermore, regarding the Key-to-Door environment mentioned in the paper, I'd be interested to see the code for that environment and the algorithm used to generate trajectories on it.