Downloads · 30 days
0
ai-week3
mharmainw/ai-week3
ai-week3 is a machine learning model from mharmainw. Use it for the machine learning task on the model card, and read the license before you ship it in a product. The card lists the license as mit.
This repository contains the best model setup for the week 3 deep learning assignment.
Access
Public
Updated Aug 10, 2026
Repo size
2.7 MB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.pt2.7 MB · 100%
At a glance
- License
- mit
- Access
- Public
- Created
- Aug 10, 2026
- Updated
- Aug 10, 2026
- SHA
- 6e932e64
Datasets
Model card
From the Hugging Face model README
AI Intern Week 03 Deep Learning
This repository contains the best model setup for the week 3 deep learning assignment.
Contents
best_model_setup.py- Defines a PyTorch neural network, loss function, and optimizer setup.
Model Setup
The model is a feed-forward neural network for 28x28 image inputs with:
- Flatten input layer
- Two hidden linear layers with ReLU activations
- Output layer with 10 classes
- Cross entropy loss
- Adam optimizer
Default training parameters:
- Learning rate:
0.001 - Epochs:
20
Usage
from best_model_setup import create_best_model_setup
model, loss_fn, optimizer = create_best_model_setup()
- License
- mit
- Languages
- en
- Created
- Aug 10, 2026
- Updated
- Aug 10, 2026