Fix load_from_tensor (#1231)

Author: araffin

docs/misc/changelog.rst

@@ -4,7 +4,7 @@ Changelog
 ==========
 
 
-Release 1.7.0a8 (WIP)
+Release 1.7.0a9 (WIP)
 --------------------------
 
 Breaking Changes:
@@ -39,6 +39,7 @@ Bug Fixes:
 - Fixed ``Self`` return type using ``TypeVar``
 - Fixed the env checker, the key was not passed when checking images from Dict observation space
 - Fixed ``normalize_images`` which was not passed to parent class in some cases
+- Fixed ``load_from_vector`` that was broken with newer PyTorch version when passing PyTorch tensor
 
 Deprecations:
 ^^^^^^^^^^^^^

stable_baselines3/common/policies.py

@@ -199,7 +199,7 @@ def load_from_vector(self, vector: np.ndarray) -> None:
 
         :param vector:
         """
-        th.nn.utils.vector_to_parameters(th.FloatTensor(vector, device=self.device), self.parameters())
+        th.nn.utils.vector_to_parameters(th.as_tensor(vector, dtype=th.float, device=self.device), self.parameters())
 
     def parameters_to_vector(self) -> np.ndarray:
         """

stable_baselines3/version.txt

@@ -1 +1 @@
-1.7.0a8
+1.7.0a9