modified a variable name

2023-04-29 07:57:05 +09:00
parent 0eb66997fb
commit e5e8bcb284
1 changed files with 2 additions and 3 deletions
--- a/tools.py
+++ b/tools.py
@@ -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