huggingface / jat

General multi-task deep RL Agent

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

BabyAI new data structure

qgallouedec opened this issue · comments

@ClementRomac I realize that datasets does not support nested dict. We will need to turn this structure

{
    "observations": {
        "mission": ...,
        "direction": ...,
        "image": ...,
        "rgb_image": ...,
    },
    "actions": ...,
    "rewards": ...,
    "dones": ...,
}

to

{
    "missions": ...,
    "directions": ...,
    "images": ...,
    "rgb_images": ...,
    "actions": ...,
    "rewards": ...,
    "dones": ...,
}

Note that I also added some "s" for consistency.

I moved the go-to dataset to the main gia-repo.

Sorry, I've missed the notification.

Ok thanks, I'll fix these in the data generation code.

#37 should fix it

@ClementRomac this one can be closed, right?