Prerit Gupta preritg

preritg / model_scratch.py

Created April 26, 2020 00:34

code to train a NN from scratch to classify dog breeds

	import os
	import torch
	from torchvision import datasets
	import torchvision.transforms as transforms

	### TODO: Write data loaders for training, validation, and test sets
	## Specify appropriate transforms, and batch_sizes

	batch_size = 32
	num_workers = 0