CoolFace
Modelpublic

Alia-Mohammed/resnet-50-finetuned-brain-tumor

sourceHugging Faceapache-2.0updated 4y agoView on Hugging Face
0likes47downloads
Model Card

<!-- This model card has been generated automatically according to the information the Trainer had access to. You should probably proofread and complete it, then remove this comment. -->

resnet-50-finetuned-brain-tumor

This model is a fine-tuned version of microsoft/resnet-50 on the imagefolder dataset. It achieves the following results on the evaluation set:

  • —Loss: 0.2757
  • —Accuracy: 0.9171

Model description

More information needed

Intended uses & limitations

More information needed

Training and evaluation data

More information needed

Training procedure

Training hyperparameters

The following hyperparameters were used during training:

  • —learning_rate: 5e-05
  • —trainbatchsize: 64
  • —evalbatchsize: 64
  • —seed: 42
  • —gradientaccumulationsteps: 4
  • —totaltrainbatch_size: 256
  • —optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
  • —lrschedulertype: linear
  • —lrschedulerwarmup_ratio: 0.1
  • —num_epochs: 15

Training results

Training LossEpochStepAccuracyValidation Loss
1.32641.0300.50351.3154
1.2222.0600.64731.2254
1.05843.0901.06680.7510
0.89774.01200.92050.8060
0.7245.01500.77400.8456
0.60256.01800.60090.8720
0.49537.02100.50390.8684
0.42528.02400.41580.8904
0.36779.02700.37050.9038
0.330510.03000.33000.9049
0.311311.03300.30530.9097
0.283512.03600.28850.9116
0.261413.03900.26060.9297
0.273514.04200.27670.9187
0.257315.04500.27570.9171

Framework versions

  • —Transformers 4.26.1
  • —Pytorch 1.13.1+cu117
  • —Datasets 2.10.0
  • —Tokenizers 0.13.2