import torch
import torchvision
import matplotlib.pyplot as plt
%matplotlib inline
from torchvision.datasets import Omniglot
import torchvision.transforms as transforms
from torch.utils.data import random_split
from torch.utils.data import DataLoader
import torch.nn as nn
import torch.nn.functional as F
import jovian

# https://pytorch.org/docs/stable/torchvision/datasets.html
dataset = Omniglot(root='data2/', download=True)

len(dataset)

# test_dataset = (root='Futurama/')
# len(test_dataset)

dataset[0]