saymrwulf/stable-baselines3
1
0
Fork 0
mirror of https://github.com/saymrwulf/stable-baselines3.git synced 2026-07-09 17:29:20 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Copy by default

Browse source
This commit is contained in:
Antonin Raffin 2020-02-03 15:42:14 +01:00
parent f7af08bea4
commit 546fbca20c
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
torchy_baselines/common/buffers.py
View file

@ -92,10 +92,10 @@ class BaseBuffer(object):
"""
raise NotImplementedError()
def to_torch(self, array: np.ndarray, copy: bool = False) -> th.Tensor:
def to_torch(self, array: np.ndarray, copy: bool = True) -> th.Tensor:
"""
Convert a numpy array to a PyTorch tensor.
Note: it does not copy the data by default
Note: it copies the data by default
:param array: (np.ndarray)
:param copy: (bool) Whether to copy or not the data