CoolFace
Modelpublic

swadhindas324/vit-convnext-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-convnext-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.6517
  • —Accuracy: 81.25
  • —Bleu-1: 0.5750
  • —Bleu-2: 0.4399
  • —Bleu-3: 0.3481
  • —Bleu-4: 0.2828
  • —Meteor: 0.5351
  • —Rouge-l: 0.4949
  • —Cider: 1.5140

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.286180.420.46680.30270.21150.15370.40860.39760.4800
1.59772.015361.225379.940.62350.48350.38450.30980.61940.56581.5422
0.84793.023041.206381.50.69090.57000.48070.40980.68760.63862.2467
0.61534.030721.353081.260.64390.51290.42010.35150.61720.57241.8687
0.61535.038401.460180.320.63300.49990.40640.33680.62270.57261.7900
0.42606.046081.526281.70.64880.51740.42540.35660.61680.57631.8769
0.34847.053761.630181.620.63900.50530.41110.34100.61360.56431.7938
0.31148.061441.651781.250.57500.43990.34810.28280.53510.49491.5140

Framework versions

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