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

Downloads last month: -; Downloads are not tracked for this model. How to track

Inference Providers NEW

This model isn't deployed by any Inference Provider. 🙋 Ask for provider support