DA6401 Assignment 1 - MLP for Image Classification

Links

• W&B Report: https://wandb.ai/kannikantipardhasaradhi-indian-institute-of-technology-m/da6401_assignment_1/reports/DA6401-Assignment-1-MLP-for-Image-Classification--VmlldzoxNjEzMDk0Mw?accessToken=n6rb2f1qzmxfugie4l8qc5p5vr4cf73ey6ehok1vhr385g6fin5zmf06priuhng9
• GitHub Repository: https://github.com/pardhgit4bit/da6401_assignment_1

How to Train

cd src
python train.py

How to Run Inference

cd src
python inference.py

Best Model Configuration

• Optimizer: RMSProp
• Activation: ReLU
• Hidden layers: 3 x 128 neurons
• Learning rate: 0.001
• Loss: Cross Entropy
• Weight init: Xavier
• Test F1: 0.9767