CoolFace
Modelpublic

swadhindas324/vit-resnet-Mistral-RSICD-without-captioning

sourceHugging Faceupdated 2mo agoView on Hugging Face
0likes3downloads
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. -->

vit-resnet-Mistral-RSICD-without-captioning

This model is a fine-tuned version of [](https://huggingface.co/) on an unknown dataset. It achieves the following results on the evaluation set:

  • —Loss: 1.6545
  • —Accuracy: 81.45
  • —Bleu-1: 0.6169
  • —Bleu-2: 0.4800
  • —Bleu-3: 0.3863
  • —Bleu-4: 0.3200
  • —Meteor: 0.5836
  • —Rouge-l: 0.5397
  • —Cider: 1.7724

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: 0.0001
  • —trainbatchsize: 64
  • —evalbatchsize: 64
  • —seed: 50
  • —optimizer: Use OptimizerNames.ADAMWTORCHFUSED with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
  • —lrschedulertype: linear
  • —lrschedulerwarmup_steps: 0.1
  • —num_epochs: 128
  • —mixedprecisiontraining: Native AMP

Training results

Training LossEpochStepValidation LossAccuracyBleu-1Bleu-2Bleu-3Bleu-4MeteorRouge-lCider
No log1.07681.288779.760.46670.30260.21140.15360.40840.39740.4796
1.59922.015361.236379.460.61920.47970.38030.30510.61870.55941.4971
0.83743.023041.223281.120.67010.54500.45350.38120.66510.61512.0863
0.60044.030721.363781.230.64970.51940.42640.35660.63170.58951.9473
0.60045.038401.456280.420.61830.48290.39080.32250.59400.54711.6977
0.42186.046081.524881.130.62810.50040.41140.34480.60800.56941.8229
0.34697.053761.635081.470.63180.50150.40770.33770.61780.56701.7947
0.31058.061441.654581.450.61690.48000.38630.32000.58360.53971.7724

Framework versions

  • —Transformers 5.12.1
  • —Pytorch 2.12.1+cu130
  • —Datasets 5.0.0
  • —Tokenizers 0.22.2