wassname/dreamerv3-torch
1
0
Fork 0
mirror of https://github.com/wassname/dreamerv3-torch.git synced 2026-06-27 17:30:36 +08:00
Code Issues Packages Projects Releases Wiki Activity

modified a variable name

Browse Source
This commit is contained in:
NM512
2023-04-29 07:57:05 +09:00
parent 0eb66997fb
commit e5e8bcb284
1 changed files with 2 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tools.py
+2 -3
View File
@@ -328,9 +328,8 @@ class TwoHotDistSymlog:
self.width = (self.buckets[-1] - self.buckets[0]) / 255
def mean(self):
print("mean called")
_mode = self.probs * self.buckets
return symexp(torch.sum(_mode, dim=-1, keepdim=True))
_mean = self.probs * self.buckets
return symexp(torch.sum(_mean, dim=-1, keepdim=True))
def mode(self):
_mode = self.probs * self.buckets