CoolFace
Modelpublic

edlee123/bridgetower

sourceHugging Facemitupdated 2y agoView on Hugging Face
1likes12downloads
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. -->

bridgetower

This model is a fine-tuned version of BridgeTower/bridgetower-large-itm-mlm-itc on the jmhessel/newyorkercaptioncontest matching dataset. It achieves the following results on the evaluation set:

  • —Loss: 0.1309
  • —Memory Allocated (gb): 51.27
  • —Max Memory Allocated (gb): 57.18
  • —Total Memory Available (gb): 94.62

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: 1e-05
  • —trainbatchsize: 40
  • —evalbatchsize: 16
  • —seed: 42
  • —optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-06
  • —lrschedulertype: linear
  • —num_epochs: 8.0

Training results

Framework versions

  • —Transformers 4.43.4
  • —Pytorch 2.3.1a0+git4989238
  • —Datasets 3.0.1
  • —Tokenizers 0.19.1