added README.md

README.md

@@ -0,0 +1,24 @@
+---
+datasets:
+- cifar10
+metrics:
+- accuracy
+library_name: pytorch
+pipeline_tag: image-classification
+---
+
+# custom-cnn-cifar2
+
+Custom convolutional neural network (CNN) trained on CIFAR-2 (a subset of CIFAR-10 for classifying 'airplane' vs. 'bird').
+
+This model pertains to Exercise 1 of Chapter 8 of the book "Deep Learning with PyTorch" by Eli Stevens, Luca Antiga, and Thomas Viehmann.
+
+**Note:** In the exercise, we tried out `(5, 5)` and `(1, 3)` convolution kernel sizes. However, these didn't outperform the baseline network with `(3, 3)` kernel size. Hence, this checkpoint sticks to the `(3, 3)` kernel size.
+
+Code: https://github.com/sambitmukherjee/dlwpt-exercises/blob/main/chapter_8/exercise_1.ipynb
+
+Experiment tracking: https://wandb.ai/sadhaklal/custom-cnn-cifar2
+
+## Metric
+
+Accuracy on `cifar2_val`: 0.8995