How do I set the size of the Neural Network to [16, 16] in sb3?

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

942 stars 69 forks source link

How do I set the size of the Neural Network to [16, 16] in sb3? #32

Closed ryash072007 closed 1 year ago

ryash072007 commented 1 year ago

I noticed that the size is set to [64, 64] and changing it in the config file does nothing. Btw, I managed to get my custom environment up and running and it works! Asking this question becoz it doesn't need a complex and large brain.

edbeeching commented 1 year ago

I have added a separate example file to this repo for sb3, examples/stable_baselines3_example.py . You can find more info on how to configure the size of the nn in the sb3 docs here