AI417UPM/A4_4320192_Enas

AI417UPM - Assignment 4

Overview

This repository contains two neural network architectures implemented using PyTorch:

• DummyNet
• VanillaNet

The models are part of Assignment 4 for AI417 and can be loaded using PyTorch Hub.

Repository Contents

README.md

dummy-weights.bin

dummy.py

hubconf.py

vanilla-weights.bin

vanilla.py

Dataset

The models were trained using the FashionMNIST dataset, which contains grayscale images of clothing items such as shirts, shoes, and bags.

Models Description

dummy.py

• The DummyNet model is a linear feedforward neural network.
• It does not contain activation functions.

---

vanilla.py

• It includes activation functions to improve learning capability and model performance.

Model Weights

This repository contains pretrained weights:

• dummy-weights.bin → Weights for DummyNet.
• vanilla-weights.bin → Weights for VanillaNet.

These weights can be loaded using PyTorch Hub.