fix typo

2026-06-05 14:01:09 -04:00
parent 6eaf8bea69
commit c252961635
1 changed files with 1 additions and 1 deletions
@@ -132,7 +132,7 @@ def _make_call_model_fn(model, class_idx):
                images_np = np.array(images, dtype=np.float32)
                images_tensor = torch.from_numpy(images_np).permute(0, 3, 1, 2)

-                mmean = torch.tensor(IMAGENET_MEAN).view(1, 3, 1, 1).to(DEVICE)
+                mean = torch.tensor(IMAGENET_MEAN).view(1, 3, 1, 1).to(DEVICE)
                std  = torch.tensor(IMAGENET_STD).view(1, 3, 1, 1).to(DEVICE)
                images_tensor = images_tensor.to(DEVICE)
                images_tensor = (images_tensor - mean) / std