README.md

metadata

license: mit

AI417 – Assignment 4 (Swiss Knife)

This repository contains two PyTorch models submitted for AI417 Assignment 4.

Models

• DummyNet: a simple feed-forward network
• VanillaNet: a deeper feed-forward network with LeakyReLU activations

Usage

Models are compatible with torch.hub.load and FashionMNIST-sized inputs (784).

Files

• Dummy.py – DummyNet architecture
• Vanilla.py – VanillaNet architecture
• hubconf.py – HuggingFace hub configuration
• dummy_weights.bin – DummyNet weights
• vanilla_weights.bin – VanillaNet weights