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

fix #85

Browse Source
This commit is contained in:
Trinkle23897 2020-06-22 17:11:26 +08:00
parent a655334d00
commit 6a2963bd64
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tianshou/policy/base.py
View File

@ -125,7 +125,7 @@ class BasePolicy(ABC, nn.Module):
:return: a Batch. The result will be stored in batch.returns. :return: a Batch. The result will be stored in batch.returns.
""" """
if v_s_ is None: if v_s_ is None:
v_s_ = np.zeros_like(batch.rew) v_s_ = batch.rew * 0.
else: else:
if not isinstance(v_s_, np.ndarray): if not isinstance(v_s_, np.ndarray):
v_s_ = np.array(v_s_, np.float) v_s_ = np.array(v_s_, np.float)