Official release of the datasets from the paper "Hierarchical Entity-centric Reinforcement Learning with Factored Subgoal Diffusion" by Dan Haramati, Carl Qi,
Tal Daniel, Amy Zhang, Aviv Tamar and George Konidaris.
Datasets contain 128x128 RGB images, state information, actions, and information for calculating goal-conditioned rewards.
For RL… See the full description on the dataset page: https://huggingface.co/datasets/DanHrmti/hecrl.