AI417UPM/A4_4312595_Tithkar

AI417 – Assignment 4

Name: Tithkar
UPM ID: 4312595

---

Models Included

This repository contains two PyTorch models for FashionMNIST classification (input size = 784, output classes = 10).

1️⃣ DummyNet

• File: dummy.py
• Weights: dummy_weights.bin

2️⃣ VanillaNet

• File: vanilla.py
• Weights: vanilla_weights.bin

---

Usage (Torch Hub)

The models can be loaded using:

import torch as tr
from huggingface_hub import snapshot_download

repo = snapshot_download("AI417UPM/A4_4312595_Tithkar")

model = tr.hub.load(repo, "DummyNet", source="local")
model.eval()

x = tr.randn(1, 784)
y = model(x)
print(y.shape)  # torch.Size([1, 10])