Learn practical skills, build real-world projects, and advance your career
Recommended to use Kaggle Kernel
import torch
import torchvision
import numpy as np
import matplotlib.pyplot as plt
import torch.nn as nn
import torch.nn.functional as F
from torchvision.transforms import ToTensor
from torchvision.datasets import MNIST
from torch.utils.data.sampler import SubsetRandomSampler
from torch.utils.data.dataloader import DataLoader
dataset = MNIST(root = 'data/',
               transform = ToTensor())

Preparing the data

def split_indices(n, val_frac):
    n_value = int(n * val_frac)
    idxs = np.random.permutation(n)
    return idxs[n_value:], idxs[:n_value] # for train_data and validation_data respectively