autocrop-music

This model is a fine-tuned version of nvidia/mit-b0 on the /home/nbspark/treningsdata/music dataset. It achieves the following results on the evaluation set:

  • Loss: 0.0130
  • Mean Iou: 0.4907
  • Mean Accuracy: 0.9814
  • Overall Accuracy: 0.9814
  • Accuracy Background: nan
  • Accuracy Crop: 0.9814
  • Iou Background: 0.0
  • Iou Crop: 0.9814

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

Training results

Training Loss Epoch Step Validation Loss Mean Iou Mean Accuracy Overall Accuracy Accuracy Background Accuracy Crop Iou Background Iou Crop
No log 1.0 75 0.4329 0.4821 0.9642 0.9642 nan 0.9642 0.0 0.9642
No log 2.0 150 0.1566 0.4787 0.9574 0.9574 nan 0.9574 0.0 0.9574
No log 3.0 225 0.1025 0.4753 0.9505 0.9505 nan 0.9505 0.0 0.9505
No log 4.0 300 0.0740 0.4699 0.9397 0.9397 nan 0.9397 0.0 0.9397
No log 5.0 375 0.0588 0.4799 0.9597 0.9597 nan 0.9597 0.0 0.9597
No log 6.0 450 0.0456 0.4835 0.9669 0.9669 nan 0.9669 0.0 0.9669
0.2228 7.0 525 0.0346 0.4847 0.9694 0.9694 nan 0.9694 0.0 0.9694
0.2228 8.0 600 0.0290 0.4795 0.9591 0.9591 nan 0.9591 0.0 0.9591
0.2228 9.0 675 0.0286 0.4877 0.9754 0.9754 nan 0.9754 0.0 0.9754
0.2228 10.0 750 0.0245 0.4866 0.9733 0.9733 nan 0.9733 0.0 0.9733
0.2228 11.0 825 0.0225 0.4891 0.9782 0.9782 nan 0.9782 0.0 0.9782
0.2228 12.0 900 0.0313 0.4958 0.9916 0.9916 nan 0.9916 0.0 0.9916
0.2228 13.0 975 0.0212 0.4924 0.9848 0.9848 nan 0.9848 0.0 0.9848
0.0386 14.0 1050 0.0207 0.4913 0.9826 0.9826 nan 0.9826 0.0 0.9826
0.0386 15.0 1125 0.0182 0.4924 0.9848 0.9848 nan 0.9848 0.0 0.9848
0.0386 16.0 1200 0.0210 0.4945 0.9890 0.9890 nan 0.9890 0.0 0.9890
0.0386 17.0 1275 0.0200 0.4814 0.9628 0.9628 nan 0.9628 0.0 0.9628
0.0386 18.0 1350 0.0174 0.4873 0.9745 0.9745 nan 0.9745 0.0 0.9745
0.0386 19.0 1425 0.0178 0.4910 0.9820 0.9820 nan 0.9820 0.0 0.9820
0.0232 20.0 1500 0.0169 0.4923 0.9846 0.9846 nan 0.9846 0.0 0.9846
0.0232 21.0 1575 0.0157 0.4904 0.9807 0.9807 nan 0.9807 0.0 0.9807
0.0232 22.0 1650 0.0162 0.4921 0.9841 0.9841 nan 0.9841 0.0 0.9841
0.0232 23.0 1725 0.0152 0.4892 0.9785 0.9785 nan 0.9785 0.0 0.9785
0.0232 24.0 1800 0.0150 0.4905 0.9810 0.9810 nan 0.9810 0.0 0.9810
0.0232 25.0 1875 0.0153 0.4918 0.9836 0.9836 nan 0.9836 0.0 0.9836
0.0232 26.0 1950 0.0151 0.4877 0.9755 0.9755 nan 0.9755 0.0 0.9755
0.0183 27.0 2025 0.0144 0.4901 0.9802 0.9802 nan 0.9802 0.0 0.9802
0.0183 28.0 2100 0.0144 0.4888 0.9777 0.9777 nan 0.9777 0.0 0.9777
0.0183 29.0 2175 0.0146 0.4922 0.9843 0.9843 nan 0.9843 0.0 0.9843
0.0183 30.0 2250 0.0140 0.4907 0.9813 0.9813 nan 0.9813 0.0 0.9813
0.0183 31.0 2325 0.0141 0.4936 0.9872 0.9872 nan 0.9872 0.0 0.9872
0.0183 32.0 2400 0.0136 0.4913 0.9826 0.9826 nan 0.9826 0.0 0.9826
0.0183 33.0 2475 0.0135 0.4920 0.9840 0.9840 nan 0.9840 0.0 0.9840
0.0158 34.0 2550 0.0137 0.4928 0.9856 0.9856 nan 0.9856 0.0 0.9856
0.0158 35.0 2625 0.0136 0.4926 0.9852 0.9852 nan 0.9852 0.0 0.9852
0.0158 36.0 2700 0.0140 0.4929 0.9859 0.9859 nan 0.9859 0.0 0.9859
0.0158 37.0 2775 0.0131 0.4920 0.9841 0.9841 nan 0.9841 0.0 0.9841
0.0158 38.0 2850 0.0132 0.4916 0.9832 0.9832 nan 0.9832 0.0 0.9832
0.0158 39.0 2925 0.0134 0.4920 0.9840 0.9840 nan 0.9840 0.0 0.9840
0.0142 40.0 3000 0.0131 0.4925 0.9849 0.9849 nan 0.9849 0.0 0.9849
0.0142 41.0 3075 0.0131 0.4926 0.9853 0.9853 nan 0.9853 0.0 0.9853
0.0142 42.0 3150 0.0132 0.4920 0.9840 0.9840 nan 0.9840 0.0 0.9840
0.0142 43.0 3225 0.0132 0.4923 0.9847 0.9847 nan 0.9847 0.0 0.9847
0.0142 44.0 3300 0.0132 0.4924 0.9847 0.9847 nan 0.9847 0.0 0.9847
0.0142 45.0 3375 0.0132 0.4921 0.9841 0.9841 nan 0.9841 0.0 0.9841
0.0142 46.0 3450 0.0132 0.4921 0.9842 0.9842 nan 0.9842 0.0 0.9842
0.0141 47.0 3525 0.0132 0.4921 0.9843 0.9843 nan 0.9843 0.0 0.9843
0.0141 48.0 3600 0.0132 0.4920 0.9840 0.9840 nan 0.9840 0.0 0.9840
0.0141 49.0 3675 0.0131 0.4922 0.9844 0.9844 nan 0.9844 0.0 0.9844
0.0141 50.0 3750 0.0130 0.4907 0.9814 0.9814 nan 0.9814 0.0 0.9814

Framework versions

  • Transformers 5.8.0
  • Pytorch 2.11.0+cu130
  • Datasets 4.8.5
  • Tokenizers 0.22.2
Downloads last month
25
Safetensors
Model size
3.72M params
Tensor type
F32
·
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Model tree for gkberg/autocrop-music

Base model

nvidia/mit-b0
Finetuned
(460)
this model