AI Intern Week 03 Deep Learning

Contents

• best_model_setup.py - Defines a PyTorch neural network, loss function, and optimizer setup.

Model Setup

• Flatten input layer
• Two hidden linear layers with ReLU activations
• Output layer with 10 classes
• Cross entropy loss
• Adam optimizer

• Learning rate: 0.001
• Epochs: 20

Usage