Remove print in MNIST dataset

434271dc · Adam Paszke · 80b38a07 · 434271dc
Commit 434271dc authored Jan 19, 2017 by Adam Paszke
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 3 deletions

torchvision/datasets/mnist.py torchvision/datasets/mnist.py +2 -3

No files found.
--- a/torchvision/datasets/mnist.py
+++ b/torchvision/datasets/mnist.py
@@ -72,7 +72,6 @@ class MNIST(data.Dataset):
        import gzip

        if self._check_exists():
-            print('Files already downloaded')
            return

        # download files
@@ -98,8 +97,8 @@ class MNIST(data.Dataset):
            os.unlink(file_path)

        # process and save as torch files
-        print('Processing')
-        
+        print('Processing...')
+
        training_set = (
            read_image_file(os.path.join(self.root, self.raw_folder, 'train-images-idx3-ubyte')),
            read_label_file(os.path.join(self.root, self.raw_folder, 'train-labels-idx1-ubyte'))