You can deploy this model and start earning money today!

AI417UPM/
A4_4311950_Sana

Views

No views yet

A4 - AI417 TorchHub Models

Student Name: Sana Faisal
Student ID: 4311950

Models Included

This repository contains two PyTorch models:

DummyNet
VanillaNet

Both models are simple feedforward neural networks trained on FashionMNIST.

Files

dummy.py → Dummy model architecture
vanilla.py → Vanilla model architecture
dummy_weights.bin → Trained weights for DummyNet
vanilla_weights.bin → Trained weights for VanillaNet
hubconf.py → TorchHub entry file

Usage

Example usage:

python

from huggingface_hub import snapshot_download
import torch as tr

repo = snapshot_download("ai417upm/A4_4311950_Sana")

model = tr.hub.load(repo, "DummyNet", source="local")
model.eval()
output = model(tr.randn(2, 784))

model = tr.hub.load(repo, "VanillaNet", source="local")
model.eval()
output = model(tr.randn(2, 784))

AI417UPM/A4_4311950_Sana

A4 - AI417 TorchHub Models

Models Included

Files

Usage

AI417UPM/
A4_4311950_Sana