refine py3 for LRC (#8)

LRC/reader_cifar.py

@@ -127,7 +127,10 @@ def reader_creator_filepath(filename, sub_name, is_training, args):
     datasets = []
     for name in names:
         print("Reading file " + name)
-        batch = pickle.load(open(filename + name, 'rb'))
+        try:
+            batch = pickle.load(open(filename + name, 'rb'), encode='latin1')
+        except TypeError:
+            batch = pickle.load(open(filename + name, 'rb'))
         data = batch['data']
         labels = batch.get('labels', batch.get('fine_labels', None))
         assert labels is not None

LRC/voting.py

@@ -12,6 +12,8 @@ pred = np.zeros((10000, 10))
 fl.sort()
 i = 0
 for f in fl:
+    if 'init' in f:
+        continue
     print(f)
     pred += pickle.load(open(os.path.join(model_path, f)))
     print(np.mean(np.argmax(pred, axis=1) == labels))