added README.md

README.md

@@ -0,0 +1,24 @@
+---
+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