Small fix in results/plot_results.py

.gitignore

@@ -0,0 +1,8 @@
+*.pt
+*.txt
+*.pyc
+*.pdf
+*.tar.gz
+*.npz
+
+/data/raw/cifar-100-python

data/raw/raw.py

@@ -8,13 +8,22 @@
 import subprocess
 import pickle
 import torch
+import os
+
+cifar_path = "cifar-100-python.tar.gz"
+mnist_path = "mnist.npz"
+
+print(os.path.exists(cifar_path))
+
+if not os.path.exists(cifar_path):
+    # URL from: https://www.cs.toronto.edu/~kriz/cifar.html
+    subprocess.call("wget https://www.cs.toronto.edu/~kriz/cifar-100-python.tar.gz", shell=True)
 
-# URL from: https://www.cs.toronto.edu/~kriz/cifar.html
-subprocess.call("wget https://www.cs.toronto.edu/~kriz/cifar-100-python.tar.gz", shell=True)
 subprocess.call("tar xzfv cifar-100-python.tar.gz", shell=True)
 
 # URL from: https://github.com/fchollet/keras/blob/master/keras/datasets/mnist.py
-subprocess.call("wget https://s3.amazonaws.com/img-datasets/mnist.npz", shell=True)
+if not os.path.exists(mnist_path):
+    subprocess.call("wget https://s3.amazonaws.com/img-datasets/mnist.npz", shell=True)
 
 def unpickle(file):
     with open(file, 'rb') as fo:

results/plot_results.py

@@ -49,7 +49,7 @@
         matches = glob(model + '*' + dataset + '*.pt')
         if len(matches):
             data = torch.load(matches[0], map_location=lambda storage, loc: storage)
-            acc, bwt, fwt = data[3][5:]
+            acc, bwt, fwt = data[3][:]
             barplot[dataset][model]['acc'] = acc
             barplot[dataset][model]['bwt'] = bwt
             barplot[dataset][model]['fwt'] = fwt