Learn practical skills, build real-world projects, and advance your career
Updated 4 years ago
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