You can deploy this model and start earning money today!

AI417UPM/
A4_4312595_Tithkar

Name: A4_4312595_Tithkar
Availability: InStock

Views

No views yet

AI417 – Assignment 4

Name: Tithkar
UPM ID: 4312595

Models Included

This repository contains two PyTorch models for FashionMNIST classification (input size = 784, output classes = 10).

1️⃣ DummyNet

File: dummy.py
Weights: dummy_weights.bin

2️⃣ VanillaNet

File: vanilla.py
Weights: vanilla_weights.bin

Usage (Torch Hub)

The models can be loaded using:

python

import torch as tr
from huggingface_hub import snapshot_download

repo = snapshot_download("AI417UPM/A4_4312595_Tithkar")

model = tr.hub.load(repo, "DummyNet", source="local")
model.eval()

x = tr.randn(1, 784)
y = model(x)
print(y.shape)  # torch.Size([1, 10])