CoolFace
Modelpublic

aee4/G17-AMFU-Net

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

G17-AMFU-Net

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: 0.2900
  • —Dice: 0.8320
  • —Iou: 0.7535
  • —Precision: 0.8675
  • —Recall: 0.8433
  • —Specificity: 0.9870

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.0003
  • —trainbatchsize: 4
  • —evalbatchsize: 4
  • —seed: 42
  • —gradientaccumulationsteps: 2
  • —totaltrainbatch_size: 8
  • —optimizer: Use OptimizerNames.ADAMWTORCH with betas=(0.9,0.999) and epsilon=1e-08 and optimizerargs=No additional optimizer arguments
  • —lrschedulertype: linear
  • —num_epochs: 60

Training results

Training LossEpochStepDiceIouValidation LossPrecisionRecallSpecificity
1.07181.18482500.58600.45031.03820.53310.82210.9518
0.79472.36975000.64670.52070.74400.59220.84470.9562
0.54983.55457500.72860.62040.56690.72350.82630.9778
0.53684.739310000.74520.64780.51510.84790.74000.9908
0.42435.924212500.76080.65990.42630.77840.81400.9821
0.42627.109015000.77460.67790.39800.78910.83360.9797
0.37258.293817500.75670.66970.43040.85970.75700.9904
0.38219.478720000.78550.69660.37790.84070.80490.9869
0.320810.663522500.78720.69510.37350.83210.80710.9869
0.42811.848325000.79780.70730.35020.83260.81800.9865
0.389713.033227500.78560.69260.37380.78790.83890.9814
0.361814.218030000.77870.68460.38710.88530.74130.9920
0.31515.402832500.80450.71870.34670.85670.81010.9890
0.379816.587735000.81160.72240.33370.82300.84810.9849
0.334517.772537500.81090.72810.32440.87500.81280.9891
0.367818.957340000.81330.72710.32780.82810.85260.9846
0.324720.142242500.81070.72520.32160.82910.85040.9805
0.347421.327045000.79560.70960.35340.87630.78650.9921
0.249622.511847500.81530.73050.32270.85560.82240.9883
0.325623.696750000.81390.72420.32350.80410.87160.9811
0.300724.881552500.81700.73140.31640.83680.84990.9834
0.314726.066455000.81640.72960.31320.84300.83770.9867
0.3227.251257500.81330.73200.32680.85510.82340.9888
0.307528.436060000.81470.72840.32800.83230.85030.9856
0.311529.620962500.81610.72990.31490.85800.82240.9845
0.28930.805765000.82760.74380.29710.86770.82990.9867
0.309631.990567500.83310.74910.28920.82730.87920.9803
0.256333.175470000.82900.74310.29980.83910.85670.9843
0.264734.360272500.81120.72750.33410.89520.78750.9916
0.248935.545075000.81910.73310.30960.82940.85400.9829
0.25236.729977500.83020.74840.29650.85910.84470.9872
0.257537.914780000.81880.73720.30770.86030.82270.9882
0.273339.099582500.82600.74610.29840.87150.82880.9887
0.268640.284485000.83070.74530.29350.84040.86340.9815
0.241341.469287500.82850.74460.29480.84380.85980.9846
0.227742.654090000.83040.75030.29720.88200.82980.9901
0.246843.838992500.83310.75150.29890.87910.83010.9897
0.249145.023795000.83070.74990.29670.87340.83940.9889
0.219946.208597500.83820.75650.28610.85980.85280.9866
0.246347.3934100000.83970.75700.28220.84880.87030.9837
0.228548.5782102500.83480.75390.28210.87150.83760.9878
0.236749.7630105000.83390.75320.29200.86510.84130.9881
0.238750.9479107500.83660.75580.28300.85140.85800.9858
0.238452.1327110000.83820.75690.27680.85970.85370.9870
0.210853.3175112500.83660.75700.28350.86620.84990.9873
0.193854.5024115000.83060.75180.29250.87590.83820.9881
0.220355.6872117500.83590.75650.28730.87670.84050.9886
0.185856.8720120000.83420.75520.29180.87710.83690.9881
0.252358.0569122500.29050.83190.75330.86590.84510.9870
0.22159.2417125000.29000.83200.75350.86750.84330.9870

Framework versions

  • —Transformers 4.57.1
  • —Pytorch 2.10.0+cu128
  • —Datasets 4.8.5
  • —Tokenizers 0.22.2