性能微小优化

fangwei123456 · Sep 8, 2020 · d8225c7 · d8225c7
1 parent 1817db5
commit d8225c7
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/spikingjelly/datasets/n_mnist.py b/spikingjelly/datasets/n_mnist.py
@@ -55,9 +55,8 @@ def read_bin(file_name: str):
             raw_data = np.uint32(np.fromfile(bin_f, dtype=np.uint8))
             x = raw_data[0::5]
             y = raw_data[1::5]
-            rd_2__5 = raw_data[2::5]
-            p = (rd_2__5 & 128) >> 7  # bit 7
-            t = ((rd_2__5 & 127) << 16) | (raw_data[3::5] << 8) | (raw_data[4::5])
+            p = (raw_data[2::5] & 128) >> 7  # bit 7
+            t = ((raw_data[2::5] & 127) << 16) | (raw_data[3::5] << 8) | (raw_data[4::5])
         return {'t': t, 'x': x, 'y': y, 'p': p}
 
     @staticmethod