zalando-datasets/fashion_mnist
Viewer • Updated • 70k • 28.3k • 67
A small “plug-and-play” TorchHub repo that exposes two models trained on FashionMNIST:
dummyNet → linear-only baseline vanillaNet → MLP with non-linearityfrom huggingface_hub import snapshot_download
import torch as tr
repo_dir = snapshot_download("AI417UPM/A4_4312330_Anas")
dummy = tr.hub.load(repo_dir, "dummyNet", source="local", pretrained=True).eval()
vanilla = tr.hub.load(repo_dir, "vanillaNet", source="local", pretrained=True).eval()
x = tr.rand(1, 784)
print("dummy logits:", dummy(x).shape) # torch.Size([1, 10])
print("vanilla logits:", vanilla(x).shape) # torch.Size([1, 10])