From 7334559751a00b71c4d510adab01f585c93af288 Mon Sep 17 00:00:00 2001
From: Eljan Mahammadli <eljanmahammadli@gmail.com>
Date: Mon, 20 Nov 2023 04:14:13 -0500
Subject: [PATCH] fixed bug: the imagenet index has to start from 0

---
 examples/imagenet.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/examples/imagenet.py b/examples/imagenet.py
index dff78a3..e23cebf 100644
--- a/examples/imagenet.py
+++ b/examples/imagenet.py
@@ -1000,4 +1000,4 @@
     "ear",
     "toilet tissue",
 ]
-IMAGENET_CATEGORIES = {i + 1: v for i, v in enumerate(IMAGENET_CATEGORIES)}
+IMAGENET_CATEGORIES = {i: v for i, v in enumerate(IMAGENET_CATEGORIES)}