From e5e8bcb2843bebe60798cbb10abbcfee4b6e35e0 Mon Sep 17 00:00:00 2001 From: NM512 Date: Sat, 29 Apr 2023 07:57:05 +0900 Subject: [PATCH] modified a variable name --- tools.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/tools.py b/tools.py index 3b8a912..24e0e3d 100644 --- 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