do not scale mnist label (#1589)

pyro-ppl · May 18, 2023 · d897d7a · d897d7a
1 parent db2913b
commit d897d7a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/numpyro/examples/datasets.py b/numpyro/examples/datasets.py
@@ -164,7 +164,7 @@ def _load_mnist():
     def read_label(file):
         with gzip.open(file, "rb") as f:
             f.read(8)
-            data = np.frombuffer(f.read(), dtype=np.int8) / np.float32(255.0)
+            data = np.frombuffer(f.read(), dtype=np.int8)
             return data
 
     def read_img(file):