-
BabyAI levels
The BabyAI levels dataset is a collection of levels with increasing degrees of complexity, where agents need to solve tasks in a simulated environment. -
OpenAI Gym
The dataset used in the paper is not explicitly described, but it is mentioned that the authors used several continuous control environments from the OpenAI Gym.