edbeeching / godot_rl_agents

An Open Source package that allows video game creators, AI researchers and hobbyists the opportunity to learn complex behaviors for their Non Player Characters or agents
MIT License
902 stars 63 forks source link

Update sb3_imitation.py #180

Closed Ivan-267 closed 5 months ago

Ivan-267 commented 5 months ago

Adds BC stage, sets tensorboard logging, adjusts hyperparams (adjusted for WIP IL tutorial env and BC pre-training case, may need manual adjustments for gail only training or etc.).

Ivan-267 commented 5 months ago

I am still testing for optimal hyperparams, it seems the last training isn't working perfectly yet, so there might be small changes after more testing. Feel free to review, however, I don't expect to make any major changes to the script.

Edit: Should be ready now.