cancel
Showing results for 
Search instead for 
Did you mean: 
Get Started Discussions
Start your journey with Databricks by joining discussions on getting started guides, tutorials, and introductory topics. Connect with beginners and experts alike to kickstart your Databricks experience.
cancel
Showing results for 
Search instead for 
Did you mean: 

Joblib with optuna and SB3

mancosta
New Contributor

Hi everyone,

I am training some reinforcement learning models and I am trying to automate the hyperparameter search using optuna. I saw in the documentation that you can use joblib with spark as a backend to train in paralel. I got that working with the example using sklearn, but now that I tried training my model using stable baselines 3 it doesn't seem to be working. Do you know if it is just not possible to do it, or is there a special way to train these models in parallel. I did not want to use Ray yet because SB3 has a lot more models out of the box than RLlib.

Thanks in advance!

0 REPLIES 0

Join Us as a Local Community Builder!

Passionate about hosting events and connecting people? Help us grow a vibrant local community—sign up today to get started!

Sign Up Now