Antonin RAFFIN

icon indicating a website link https://araffin.github.io/

icon company @DLR-RM icon location Munich icon location Research Engineer in Robotics and Machine Learning, with a focus on Reinforcement Learning.

Results 880 comments of Antonin RAFFIN

[Question] Getting a single environment from a vectorized environment save file

Hello, why would you want to do that? and what is wrong with a VecEnv that has only one env ?

[Question] Getting a single environment from a vectorized environment save file

I see, did you know that SB3 predict work with gym env too? edit: btw, vecenv resets auromatically

[Question] Getting a single environment from a vectorized environment save file

> I see, did you know that SB3 predict work with gym env too? What I mean is that we autodetect the shape of the input and output the correct...

Use MPS device when available

> Is there any progress on this? Is mps usable in any way already? @BasLaa you can already give it a try by passing `device="mps"` to the constructor and using...

Use MPS device when available

@qgallouedec how is the support with PyTorch 2.1.0?

Use MPS device when available

> Is double precision a feature of sb3 or should single precision be forced systematically? I think we don't really support float64... (mainly to avoid issues when using CUDA) there...

Use MPS device when available

> Any news regarding this PR? Is someone working on it? https://github.com/DLR-RM/stable-baselines3/pull/951#issuecomment-1493929687

[Question] C++ Inference

Closing that one in favor of another one that gonna be opened soon (per discussion with Grégoire).

Supporting PyTorch GPU compatibility on Apple Silicon chips

> Any updates on this feature request? Is it possible to use MPS with stable baselines3 now? Please have a look at the PR and the other comments, you can...

[Bug] get_obs_shape returns wrong shape for MultiBinary spaces

Should be fixed in https://github.com/DLR-RM/stable-baselines3/pull/1179