mila-iqia / babyai

BabyAI platform. A testbed for training agents to understand and execute language commands.
BSD 3-Clause "New" or "Revised" License
700 stars 146 forks source link

Mission not accomplished in the last observation #115

Open Valaybundele opened 2 years ago

Valaybundele commented 2 years ago

Hi,

I was visualising the last observation of each demonstration generated by the bot and I found that the goal was not achieved in the last observation too. Are the goals not achieved in all the generated demonstrations or am I making some mistake here? Please let me know. I used the following code for visualising the last obs in the demo. I have also attached some of the images of the last observationns for the "GoToRedBall" level. Most of them also don't have the red ball so I am just wondering if I am doing something wrong.

obs = obs_traj[-1,:,:,:] r=env.get_obs_render(obs.cpu().numpy(), 128) cv2.imwrite(f'./images/textImage.png',r)

textImage65 textImage399

Thanks