5 Commits

Author	SHA1	Message	Date
Laurent Mazare	ad73e93da2	Detach the tensors on batch-norm eval. (#1702) ... * Detach the tensors on batch-norm eval. * Fix pyo3 bindings. * Black tweak. * Formatting. * Also update the pyo3-onnx formatting. * Apply black.	2024-02-13 14:26:32 +01:00
s-casci	51e577a682	Add Policy Gradient to Reinforcement Learning examples (#1500) ... * added policy_gradient, modified main, ddpg and README * fixed typo in README * removed unnecessary imports * small refactor * Use clap for picking up the subcommand to run. --------- Co-authored-by: Laurent <laurent.mazare@gmail.com>	2023-12-30 09:01:29 +01:00
Laurent Mazare	1e86717bf2	Fix a couple typos (#1451) ... * Mixtral quantized instruct. * Fix a couple typos.	2023-12-17 05:20:05 -06:00
Travis Hammond	498c50348c	Add DDPG and fix Gym wrapper (#1207) ... * Fix Gym wrapper - It was returning things in the wrong order - Gym now differentiates between terminated and truncated * Add DDPG * Apply fixes * Remove Result annotations * Also remove Vec annotation * rustfmt * Various small improvements (avoid cloning, mutability, get clippy to pass, ...) --------- Co-authored-by: Travis Hammond <travis.hammond@alexanderthamm.com> Co-authored-by: Laurent <laurent.mazare@gmail.com>	2023-10-28 19:53:34 +01:00
Laurent Mazare	29c7f2565d	Add some reinforcement learning example. (#1090) ... * Add some reinforcement learning example. * Python initialization. * Get the example to run. * Vectorized gym envs for the atari wrappers. * Get some simulation loop to run.	2023-10-14 16:46:43 +01:00