DLR-RM / stable-baselines3

PyTorch version of Stable Baselines, reliable implementations of reinforcement learning algorithms.
https://stable-baselines3.readthedocs.io
MIT License
8.35k stars 1.6k forks source link

Hotfix: revert loading with `weights_only=True` #1913

Closed araffin closed 2 months ago

araffin commented 2 months ago

Description

closes #1911

and fixes issue from comment in https://github.com/DLR-RM/stable-baselines3/issues/1852#issuecomment-2065396286

Motivation and Context

Types of changes

Checklist

Note: You can run most of the checks using make commit-checks.

Note: we are using a maximum length of 127 characters per line