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.