README.md

metadata

license: mit
datasets:
  - ggsmith/Hymenoptera
metrics:
  - accuracy
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.

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)
)