Fix data_download script for TinyImageNet. Updated adversarial regula…

…rization(Shokri).

.gitignore

@@ -4,4 +4,4 @@
 *.ipynb_checkpoints
 *.pyc
 *.zip
-
+Datasets/*

Utils/data_downloaders.py

@@ -6,6 +6,10 @@
 
 def get_tiny_imagenet(datasets_dir):
 
+
+    if os.path.isdir(os.path.join(datasets_dir,'tiny-imagenet-200/val/images/')): 
+        os.rmdir(os.path.join(datasets_dir,'tiny-imagenet-200/val/images/'))
+
     if os.path.isdir(os.path.join(datasets_dir,'tiny-imagenet-200')):
         print('Tiny ImageNet already downloaded.')
         return
@@ -49,6 +53,9 @@ def get_tiny_imagenet(datasets_dir):
         old_file_name = os.path.join(datasets_dir,'tiny-imagenet-200/val/images',line[0])
         os.rename(old_file_name, new_file_name)
 
+
+
+
     print('Tiny ImageNet successfully downloaded and preprocessed.')
 
 

Utils/train.py

@@ -33,8 +33,10 @@ def train(net, data_loader, test_loader, optimizer, criterion, n_epochs, classes
 
         # evaluate performance on testset at the end of each epoch
         print("[%d/%d]" %(epoch, n_epochs))
+        print("Training:")
+        eval_target_net(net, data_loader, classes=classes)
+        print("Test:")
         eval_target_net(net, test_loader, classes=classes)
-
         #plt.plot(losses)
         #plt.show()