Commit 434271dc authored by Adam Paszke's avatar Adam Paszke
Browse files

Remove print in MNIST dataset

parent 80b38a07
...@@ -72,7 +72,6 @@ class MNIST(data.Dataset): ...@@ -72,7 +72,6 @@ class MNIST(data.Dataset):
import gzip import gzip
if self._check_exists(): if self._check_exists():
print('Files already downloaded')
return return
# download files # download files
...@@ -98,7 +97,7 @@ class MNIST(data.Dataset): ...@@ -98,7 +97,7 @@ class MNIST(data.Dataset):
os.unlink(file_path) os.unlink(file_path)
# process and save as torch files # process and save as torch files
print('Processing') print('Processing...')
training_set = ( training_set = (
read_image_file(os.path.join(self.root, self.raw_folder, 'train-images-idx3-ubyte')), read_image_file(os.path.join(self.root, self.raw_folder, 'train-images-idx3-ubyte')),
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment