Milo Ma myljimcia
myljimcia
/ data_loader.py
Created
November 21, 2017 06:22
— forked from kevinzakka/data_loader.py
Train, Validation and Test Split for torchvision Datasets
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| # This is an example for the CIFAR-10 dataset. | |
| # There's a function for creating a train and validation iterator. | |
| # There's also a function for creating a test iterator. | |
| # Inspired by https://discuss.pytorch.org/t/feedback-on-pytorch-for-kaggle-competitions/2252/4 | |
| from utils import plot_images | |
| def get_train_valid_loader(data_dir, | |
| batch_size, | |
| augment, |