reset model weights and deleted metrics

app.css

@@ -1,4 +1,3 @@
 body {
     background-image: url("mnist-dall.jpg");
-    background-color: #cccccc;
    }

app.py

@@ -76,7 +76,7 @@ class MNISTAdversarial_Dataset(Dataset):
         return img, label
 
 class MNISTCorrupted_By_Digit(Dataset):
-    def __init__(self,transform,digit,limit=30):
+    def __init__(self,transform,digit,limit=300):
         self.transform = transform
         self.digit = digit
         corrupted_dir="./mnist_c"
@@ -114,15 +114,13 @@ class MNISTCorrupted_By_Digit(Dataset):
 
 
 
-
-
 class MNISTCorrupted(Dataset):
     def __init__(self,transform):
         self.transform = transform
         corrupted_dir="./mnist_c"
         files = [f.name for f in os.scandir(corrupted_dir)]
-        images = [np.load(os.path.join(os.path.join(corrupted_dir,f),'test_images.npy'))[:200] for f in files]
-        labels = [np.load(os.path.join(os.path.join(corrupted_dir,f),'test_labels.npy'))[:200] for f in files]
+        images = [np.load(os.path.join(os.path.join(corrupted_dir,f),'test_images.npy'))[:300] for f in files]
+        labels = [np.load(os.path.join(os.path.join(corrupted_dir,f),'test_labels.npy'))[:300] for f in files]
         self.data = np.vstack(images)
         self.labels = np.hstack(labels)
 
@@ -151,12 +149,6 @@ TRAIN_TRANSFORM = torchvision.transforms.Compose([
                                  (0.1307,), (0.3081,))
                              ])
 
-'''
-train_loader = torch.utils.data.DataLoader(
-  torchvision.datasets.MNIST('files/', train=True, download=True,
-                             transform=TRAIN_TRANSFORM),
-  batch_size=batch_size_train, shuffle=True)
-'''
 
 test_loader = torch.utils.data.DataLoader(MNISTCorrupted(TRAIN_TRANSFORM),
                                             batch_size=batch_size_test, shuffle=False)
@@ -409,7 +401,6 @@ def get_statistics():
 
 
 def main():
-    #block = gr.Blocks(css=BLOCK_CSS)
     block = gr.Blocks(css=BLOCK_CSS)
 
     with block:

metrics.json

@@ -1 +0,0 @@
-{"all": [10.55875015258789], "0": [0.0, 0.0], "1": [0.0, 0.0], "2": [0.0, 0.0], "3": [43.33333206176758, 100.0], "4": [86.66666412353516, 0.0], "5": [0.0, 0.0], "6": [0.0, 0.0], "7": [0.0, 0.0], "8": [0.0, 0.0], "9": [0.0, 0.0]}

model.pth

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:e6fb83a68fe8dca1a7a9bc9db3029071edaf292ab2c3fda48ac3661579efe873
-size 89871

optimizer.pth

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:29d01887c93ca1c9b69aee6ceb2f77c3c0db91936d5c13e92d5dfc7075bb2237
-size 89807