You can deploy this model and start earning money today!

AI417UPM/
A4_4311914_Jana

Name: A4_4311914_Jana
Availability: InStock

Views

No views yet

Assignment 4 — Swiss Knife (AI417UPM)

Student Name: Jana
Student ID: 4311914
Course/Organization: Deep Learning @ UPM (AI417UPM)

Overview

This repository contains two feed-forward neural networks trained locally on the FashionMNIST dataset:

DummyNet (dummy.py)
VanillaNet (vanilla.py)

Both models were trained locally and their weights were uploaded to this repository.

The file hubconf.py allows loading the models using torch.hub.

Repository Structure

dummy.py → Dummy network architecture
vanilla.py → Vanilla network architecture
train_dummy.py → Training script for DummyNet
train_vanilla.py → Training script for VanillaNet
dummy-weights.bin → Trained weights for DummyNet
vanilla-weight.bin → Trained weights for VanillaNet
hubconf.py → Definitions for torch.hub loading

Model Loading Instructions

The submission can be tested using the following script:

python

from huggingface_hub import snapshot_download
import torch as tr

repo = snapshot_download("AI417UPM/A4_4311914_Jana")

model = tr.hub.load(repo, "DummyNet", source="local")
model.eval()
print("Dummy loaded successfully")

model = tr.hub.load(repo, "VanillaNet", source="local")
model.eval()
print("Vanilla loaded successfully")