blizzn

devanshuDesai / mnist_dodgers.py

Last active September 10, 2024 16:23

A neural network written in PyTorch with > 99% accuracy on the MNIST dataset.

	import torch
	from torch import nn
	from tqdm import tqdm_notebook

	class CNN(nn.Module):
	def __init__(self, input_size, num_classes):
	"""
	init convolution and activation layers
	Args:
	input_size: (1,28,28)