will-maclean / sb3-burn

Implementation of stable-baselines3 in rust with burn
MIT License
11 stars 0 forks source link

Implement VecEnv #44

Open will-maclean opened 1 week ago

will-maclean commented 1 week ago

A strength of stable-baselines3 is its vector envs, which allow for multiple envs to be executed in parallel. Let's see if we can implement the same functionality.

https://github.com/DLR-RM/stable-baselines3/tree/master/stable_baselines3/common/vec_env

Components:

will-maclean commented 1 week ago

May need to act a batch_act(&mut self, obs: Vec<O>) -> Vec<A> to the Agent trait.