Jovian
Sign In
Learn practical skills, build real-world projects, and advance your career
3 years ago
!pip install jovian --upgrade --quiet
project_name = 'mnist-capsnet'

#Loading the Dataset, Data Normalization & Augumentation

import torch
import torch.nn as nn
import torch.nn.functional as F
from torchvision.datasets import MNIST
from torch.utils.data import DataLoader
import torchvision.transforms as tt