---
datasets:
- cifar10
metrics:
- accuracy
library_name: pytorch
pipeline_tag: image-classification
---

# mlp-cifar2-v2

Multi-layer perceptron (MLP) trained on CIFAR-2 (a subset of CIFAR-10 for classifying 'airplane' vs. 'bird').

`nn.BCEWithLogitsLoss` was used to train the model.

This model pertains to Exercise 2 of Chapter 7 of the book "Deep Learning with PyTorch" by Eli Stevens, Luca Antiga, and Thomas Viehmann.

Code: https://github.com/sambitmukherjee/dlwpt-exercises/blob/main/chapter_7/exercise_2.ipynb

Experiment tracking: https://wandb.ai/sadhaklal/mlp-cifar2-v2

## Metric

Accuracy on cifar2_val: 0.829