AI417UPM/A4_4410570_Nejoud

Assignment 4: PyTorch Hub Model Deployment

Student Information :

• Name: Nejoud Fahad Alharbi.
• Student ID: 4410570

Overview :

This repository contains two neural network models (DummyNet and VanillaNet) trained on the FashionMNIST dataset. These models are configured for remote loading via PyTorch Hub.

Contents :

• dummy.py: Architecture for DummyNet.
• vanilla.py: Architecture for VanillaNet.
• dummy-weights.bin: Trained weights for DummyNet.
• vanilla-weight.bin: Trained weights for VanillaNet.
• hubconf.py: Configuration file for PyTorch Hub integration.

Usage :

Models can be loaded using the following commands:

model_dummy = torch.hub.load('AI417UPM/A4_4410570_Nejoud', 'DummyNet')
model_vanilla = torch.hub.load('AI417UPM/A4_4410570_Nejoud', 'VanillaNet')