hongshaorou/dreamerv3-torch
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

bug fix when using normal_1

Browse Source
This commit is contained in:
NM512 2023-07-27 10:01:40 +09:00
parent 6924abdd3e
commit d94a719421
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
networks.py
View File

@ -804,7 +804,7 @@ class ActionHead(nn.Module):
dist = torchd.normal.Normal(torch.tanh(mean), std) dist = torchd.normal.Normal(torch.tanh(mean), std)
dist = tools.ContDist(torchd.independent.Independent(dist, 1)) dist = tools.ContDist(torchd.independent.Independent(dist, 1))
elif self._dist == "normal_1": elif self._dist == "normal_1":
x = self._dist_layer(x) mean = self._dist_layer(x)
dist = torchd.normal.Normal(mean, 1) dist = torchd.normal.Normal(mean, 1)
dist = tools.ContDist(torchd.independent.Independent(dist, 1)) dist = tools.ContDist(torchd.independent.Independent(dist, 1))
elif self._dist == "trunc_normal": elif self._dist == "trunc_normal":