Merge remote-tracking branch 'origin/master' into mozga-intel/gcd_lcm…

…_support
apache · Sep 17, 2021 · 757b182 · 757b182
2 parents d080b2e + 72e6b16
commit 757b182
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/python/mxnet/recordio.py b/python/mxnet/recordio.py
@@ -245,10 +245,11 @@ def __init__(self, idx_path, uri, flag, key_type=int):
 
     def open(self):
         super(MXIndexedRecordIO, self).open()
-        self.idx = {}
-        self.keys = []
         self.fidx = open(self.idx_path, self.flag)
-        if not self.writable:
+        if self.writable:
+            self.idx = {}
+            self.keys = []
+        elif not self.idx:
             for line in iter(self.fidx.readline, ''):
                 line = line.strip().split('\t')
                 key = self.key_type(line[0])