Update README.md

README.md

@@ -7,4 +7,25 @@ metrics:
 pipeline_tag: image-classification
 tags:
 - pytorch
----
+---
+# Model Card for Model ID
+
+Fine-tuned for image classification of Hymenoptera (Bees and Ants). Base model from PyTorch's pretrained AlexNet.
+
+```{python}
+models.alexnet(weights=models.AlexNet_Weights.IMAGENET1K_V1)
+```
+
+## Model Details
+
+```
+Sequential(
+  (0): Dropout(p=0.5, inplace=False)
+  (1): Linear(in_features=9216, out_features=4096, bias=True)
+  (2): ReLU(inplace=True)
+  (3): Dropout(p=0.5, inplace=False)
+  (4): Linear(in_features=4096, out_features=4096, bias=True)
+  (5): ReLU(inplace=True)
+  (6): Linear(in_features=4096, out_features=2, bias=True)
+)
+```