SMAI Assignment 4 β€” MLP on CIFAR-10

Multi-layer perceptron baseline. Architecture 3072 β†’ 32 β†’ 10, hidden activation relu, 98,666 parameters, trained for 100 epochs on the true labels (SGD, lr 0.05, batch size 64). Final test accuracy 0.4216, train accuracy 0.5914.

Files: model.npz (weights for the NumPy library of the assignment, W stored as (n_in, n_out)), model.pt (PyTorch state_dict of an nn.Sequential), preprocessing.npz (per-column mean and std), config.json (architecture, training settings, preprocessing).

Preprocessing: flatten the 32Γ—32Γ—3 image to 3072 values (row, column, channel order), divide by 255, subtract mean and divide by std from preprocessing.npz.

Downloads last month
26
Inference Providers NEW
This model isn't deployed by any Inference Provider. πŸ™‹ Ask for provider support

Dataset used to train Mayank-Kar/smai-a4-mlp