This repository contains my Assignment 4 submission
It includes two models implemented and trained on the FashionMNIST dataset
contents:
dummy.py – Architecture for the first model
vanilla.py – Architecture for the second model
dummy-weights.bin – Trained weights for Dummy
vanilla-weights.bin – Trained weights for Vanilla
hubconf.py – Required file for PyTorch Hub model loading
