hongshaorou/dreamerv3-torch
1
0
Fork 0
Code Issues Pull Requests Actions 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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tools.py
View File

@ -328,9 +328,8 @@ class TwoHotDistSymlog:
self.width = (self.buckets[-1] - self.buckets[0]) / 255 self.width = (self.buckets[-1] - self.buckets[0]) / 255
def mean(self): def mean(self):
print("mean called") _mean = self.probs * self.buckets
_mode = self.probs * self.buckets return symexp(torch.sum(_mean, dim=-1, keepdim=True))
return symexp(torch.sum(_mode, dim=-1, keepdim=True))
def mode(self): def mode(self):
_mode = self.probs * self.buckets _mode = self.probs * self.buckets