From dfdb2d4af73fa66bb34ea2d13202382f5cd20e2a Mon Sep 17 00:00:00 2001 From: djech2021 <58644312+djech2021@users.noreply.github.com> Date: Sat, 26 Dec 2020 21:36:02 -0700 Subject: [PATCH] Update runningstats.py --- netdissect/runningstats.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/netdissect/runningstats.py b/netdissect/runningstats.py index 91adeb7..f67e73f 100644 --- a/netdissect/runningstats.py +++ b/netdissect/runningstats.py @@ -1017,7 +1017,7 @@ def add(self, a): centered = a - batch_mean # If more than 10 billion operations, divide into batches. if self.split_batch: - sub_batch = -(-(10 << 30) // (a.shape[1] * a.shape[1]) + sub_batch = -(-(10 << 30)) // (a.shape[1] * a.shape[1]) else: sub_batch = None # Initial batch.