From a40d664e09513b0298c0e1be43123658f28943fb Mon Sep 17 00:00:00 2001 From: Fam Zheng Date: Sat, 27 Dec 2025 10:33:48 +0000 Subject: [PATCH] refactor: update common.py --- emblem5/ai/common.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/emblem5/ai/common.py b/emblem5/ai/common.py index 3d4c036..f5e26e0 100644 --- a/emblem5/ai/common.py +++ b/emblem5/ai/common.py @@ -150,7 +150,7 @@ def make_stripe_img(left, right, nstripes): return ret def predict_multi(model, transforms, images, ncells=1): - results_per_img = ncells * ncells * 2 + results_per_img = ncells * ncells ret = [] with torch.no_grad(): tensors = []