You can deploy this model and start earning money today!

AI417UPM/
A4_4210036_Yasir

Views

No views yet

A4_4210036_Yasir

Two FashionMNIST classifiers for AI417 Deep Learning - Assignment 4.

Models

DummyNet — A simple 3-layer feedforward network (784 → 512 → 128 → 10)
VanillaNet — A deeper 5-layer feedforward network with SiLU activations (784 → 2048 → 1024 → 512 → 256 → 10)

Usage

python

from huggingface_hub import snapshot_download
import torch as tr

repo = snapshot_download('AI417UPM/A4_4210036_Yasir')

# Load DummyNet
model = tr.hub.load(repo, 'DummyNet', source='local')
model.eval()
preds = model(tr.randn(10000, 784))

# Load VanillaNet
model = tr.hub.load(repo, 'VanillaNet', source='local')
model.eval()
preds = model(tr.randn(10000, 784))

Files

File	Description
`dummy.py`	DummyNet architecture
`vanilla.py`	VanillaNet architecture
`dummy-weights.bin`	Trained DummyNet weights
`vanilla-weights.bin`	Trained VanillaNet weights
`hubconf.py`	PyTorch Hub configuration