Neural Network Assignment 2: ResNeXt & DenseNet Implementation

This repository contains PyTorch implementations and reproduction results for ResNeXt and DenseNet models as part of Neural Network Assignment 2.

Features

• 🚀 Complete implementations of:
  • ResNeXt
  • DenseNet
• ⚙️ Support for both:
  • Training from scratch
  • Fine-tuning on pre-trained models
• 📊 Performance evaluation scripts
• 📈 Training logs and visualization tools

Repository Structure

neural-network-assignment2/
├── resnext/                  # ResNeXt implementation
│   ├── resnext.py            # Fine-tuning code
│   └── resnext_scratch.py    # Training-from-scratch code
├── densenet/                 # DenseNet implementation
│   ├── densenet.py           # Fine-tuning code
│   └── densenet_scratch.py   # Training-from-scratch code
├── README.md                 # Project documentation
└── requirements.txt          # Python dependencies