Raa23 comments

Repositories
Issues
Comments

Results 1 comments of


                                            Raa23

[Bug]: PyTorch 2.7 compile results in bad state_dict keys and policy fails to load

Adding a workaround: ``` python from stable_baselines3.common.save_util import load_from_zip_file, save_to_zip_file from stable_baselines3.common.type_aliases import TensorDict model_path_to_load = "model_to_load.zip" data, params, pytorch_variables = load_from_zip_file(model_path_to_load, device='cpu', load_data=True) new_params = TensorDict() for namespace in...