search

mila-iqia / babyai

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

When I run 'BabyAI-PutNext-v0', it did not work after 24 hours' training... #111

Open Amanda2024 opened 2 years ago

Amanda2024 commented 2 years ago

python script/train_rl.py --env 'BabyAI-PutNext-v0' --tb True

Should I configure other parameters?

But I change the environment to 'Open' or 'Pickup' or 'Goto', it works well.

saleml commented 2 years ago

Please use minigrid, which includes all the BabyAI environments, and is actively maintained.

For RL, use torch-ac.