import torch
import torchvision
import torchvision.transforms as transforms

train_dataset = torchvision.datasets.MNIST( root='/data/cameron/toy_1d_datasets/', train=True, download=True, transform=transforms.ToTensor())
#test_dataset = torchvision.datasets.MNIST( root='/data/cameron/toy_1d_datasets/', train=False, download=True, transform=transforms.ToTensor())

torch.save((train_dataset.data,train_dataset.targets),"/data/cameron/toy_1d_datasets/mnist.pt")