hongshaorou/Tianshou
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix to_numpy. (#73)

Browse Source 
Co-authored-by: Alexis Duburcq <alexis.duburcq@wandercraft.eu>
This commit is contained in:
Alexis DUBURCQ 2020-06-04 16:32:05 +02:00 committed by GitHub
parent 7bf202f195
commit 66be5641b6
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tianshou/data/batch.py
View File

@ -95,7 +95,7 @@ class Batch:
k__ = '_' + k + '@' + k_ k__ = '_' + k + '@' + k_
self.__dict__[k__] = v_ self.__dict__[k__] = v_
else: else:
self.__dict__[k] = kwargs[k] self.__dict__[k] = v
def __getstate__(self): def __getstate__(self):
"""Pickling interface. Only the actual data are serialized """Pickling interface. Only the actual data are serialized
@ -180,7 +180,7 @@ class Batch:
""" """
for k, v in self.__dict__.items(): for k, v in self.__dict__.items():
if isinstance(v, torch.Tensor): if isinstance(v, torch.Tensor):
self.__dict__[k] = v.cpu().numpy() self.__dict__[k] = v.detach().cpu().numpy()
elif isinstance(v, Batch): elif isinstance(v, Batch):
v.to_numpy() v.to_numpy()