search

will-maclean / sb3-burn

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

OfflineTrainer derive optimiser backend #40

Open will-maclean opened 2 weeks ago

will-maclean commented 2 weeks ago

Currently OfflineTrainer cannot derive the backend for its optimiser, meaning it needs to be defined as e.g.:

let trainer: OfflineTrainer<_, Adam<LibTorch>, _, _, _> = OfflineTrainer::new(...)

This is clunky. Fix so that OfflineTrainer can derive all its generics