corentin/torch_utils
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Fix MNIST test data loading, add named buffer summary

Browse source
This commit is contained in:
Corentin Risselin 2020-04-17 12:08:16 +09:00
commit 7db99ffa51
3 changed files with 10 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

2
dataset/mnist.py
View file

@ -61,6 +61,6 @@ def load_data(data_path: str, flatten: bool = False) -> Tuple[np.ndarray, np.nda
train_images = load_image_file(os.path.join(data_path, train_images_filename), flatten=flatten)
train_labels = load_label_file(os.path.join(data_path, train_labels_filename))
test_images = load_image_file(os.path.join(data_path, test_images_filename), flatten=flatten)
test_labels = load_label_file(os.path.join(data_path, train_labels_filename))
test_labels = load_label_file(os.path.join(data_path, test_labels_filename))
return train_images, train_labels, test_images, test_labels