Views
No views yet
dummyNet → linear-only baselinevanillaNet → MLP with non-linearity1from huggingface_hub import snapshot_download
2import torch as tr
3
4repo_dir = snapshot_download("AI417UPM/A4_4312330_Anas")
5
6dummy = tr.hub.load(repo_dir, "dummyNet", source="local", pretrained=True).eval()
7vanilla = tr.hub.load(repo_dir, "vanillaNet", source="local", pretrained=True).eval()
8
9x = tr.rand(1, 784)
10print("dummy logits:", dummy(x).shape) # torch.Size([1, 10])
11print("vanilla logits:", vanilla(x).shape) # torch.Size([1, 10])