ai417 – assignment 4 (swiss knife)
this repo contains two pytorch feed-forward models for fashionmnist (784 → 10):
- dummynet: single linear layer (784 → 10)
- vanillanet: mlp (784 → 768 → 384 → 192 → 10) with relu
files
dummy.py – dummynet architecture
vanilla.py – vanillanet architecture
hubconf.py – torch hub entrypoints for testing
dummy-weights.bin – trained weights
vanilla-weights.bin – trained weights
loading (tester)
torch.hub.load(repo_path, "DummyNet", source="local")
torch.hub.load(repo_path, "VanillaNet", source="local")
requirements
python 3.x
torch
numpy
