Edit model card

tags:

  • image-classification
  • pytorch metrics:
  • accuracy

model-index: - name: gender-classification results: - task: name: Image Classification type: image-classification metrics: - name: Accuracy type: accuracy value: 0.970833

gender-classification

Evaluate set: 240 unseen images, from 27/05-29/05 from ShotX, no duplicate, clean, 120 male and 120 female. Loss function is CrossEntropy. Model finetuning on 1827 images from 15/05-21/05 from ShotX, base on rizvandwiki/gender-classification. . Using AdamW optimizer, weight_decay=0.05, CosineAnnealingLR scheduler, learning rate 5e-6, 20 epochs.

    accuracy    loss

    0.970833    0.102212   

Example Images

female

female

male

male

Downloads last month
432
Safetensors
Model size
85.8M params
Tensor type
F32
·
Inference Examples
This model does not have enough activity to be deployed to Inference API (serverless) yet. Increase its social visibility and check back later, or deploy to Inference Endpoints (dedicated) instead.