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?