search

btx0424 / OmniDrones

https://omnidrones.readthedocs.io/
MIT License
125 stars 23 forks source link

Stable_baselines3 #69

Open Omikaa opened 2 months ago

Omikaa commented 2 months ago

Hello,can I use stable_baselines3 to work with the omnidrones platform. My algo work was on the stable_baselines3,and I want to migrate it. How can I modify the algo part in Omnidrones?

btx0424 commented 2 months ago

I don't think you should use SB3 here. Not because it is not good, but because it is not designed for a large number of parallel environments. That's also why I did not follow the old OpenAI Gym interface. I would suggest implementing your own algorithm following the torchrl interface.

Omikaa commented 2 months ago

Thanks a lot, I get your suggestion.Maybe after I finish my latest work, I will consider using omnidrones.