Dataset1-FasterViT / README.md
Nicole-M's picture
Update README.md
c48c67e verified
metadata
metrics:
  - accuracy
tags:
  - image-classification
  - breast cancer
  - fasterViT

This model is a fine-tuned version of FasterViT on the Mammogram_V1 dataset. It achieves the following results on the evaluation set:

  • Loss: 0.4760
  • Accuracy: 0.5694

Model description

More information needed

Intended uses & limitations

More information needed

Training hyperparameters

The following hyperparameters were used during training:

  • learning_rate: 5e-4
  • train_batch_size: 32
  • eval_batch_size: 32

Training and evaluation data

Training Loss Epoch Validation Loss Accuracy
0.4868 1.0 0.4745 0.4700
0.4767 2.0 0.4750 0.4711
0.4761 3.0 0.4714 0.5203
0.4746 4.0 0.4710 0.5694

Framework Versions

  • Pytorch 2.3.1+cu121
  • Datasets 2.21.0