!pip install jovian --upgrade --quiet

import torch
import jovian
import torchvision
import torch.nn as nn
import matplotlib.pyplot as plt
import torch.nn.functional as F
import torchvision.transforms as transforms
from torchvision.datasets import QMNIST
from torch.utils.data import random_split
from torch.utils.data import DataLoader
print('all imported')

# Hyperparmeters
batch_size = 128
learning_rate = 0.0001

# Other constants
input_size = 28*28
num_classes = 10

jovian.reset()
jovian.log_hyperparams(batch_size=batch_size, learning_rate=learning_rate)