AI417UPM/A4_4410891_Abdullah

Assignment 4: Swiss Knife! (Deep Learning)

Student Name: Abdullah
Student ID: 4410891

This repository contains the model architectures and pre-trained weights for Assignment 4.

Repository Structure

• dummy.py: Architecture for the Dummy network.
• vanilla.py: Architecture for the Vanilla network.
• dummy-weights.bin: Trained weights for the Dummy model.
• vanilla-weight.bin: Trained weights for the Vanilla model.
• hubconf.py: Configuration file for PyTorch Hub loading.

Usage

The models can be loaded using torch.hub as follows:

import torch as tr
# Load the Dummy model
model = tr.hub.load('path_to_repo', 'DummyNet', source='local')
# Load the Vanilla model
model = tr.hub.load('path_to_repo', 'VanillaNet', source='local')
\```