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()
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support