danielbogdoll's picture
Update README.md
4719c91 verified
metadata
library_name: transformers
base_model: jozhang97/deta-swin-large
tags:
  - generated_from_trainer
datasets:
  - Voxel51/fisheye8k
model-index:
  - name: fisheye8k_jozhang97_deta-swin-large
    results: []

fisheye8k_jozhang97_deta-swin-large

This model is a fine-tuned version of jozhang97/deta-swin-large on the generator dataset. It achieves the following results on the evaluation set:

  • Loss: 17.9701

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
  • train_batch_size: 1
  • eval_batch_size: 8
  • seed: 0
  • optimizer: Use OptimizerNames.ADAMW_TORCH with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
  • lr_scheduler_type: cosine
  • num_epochs: 36
  • mixed_precision_training: Native AMP

Training results

Training Loss Epoch Step Validation Loss
13.7551 1.0 5288 17.5573
12.6537 2.0 10576 17.4879
12.023 3.0 15864 17.6520
11.4167 4.0 21152 18.5138
10.8161 5.0 26440 17.7264
10.5346 6.0 31728 17.9145
10.1203 7.0 37016 17.9701

Framework versions

  • Transformers 4.48.3
  • Pytorch 2.5.1+cu124
  • Datasets 3.2.0
  • Tokenizers 0.21.0