End of training

README.md

@@ -16,32 +16,6 @@ should probably proofread and complete it, then remove this comment. -->
 # segformer-b0-finetuned-v0
 
 This model is a fine-tuned version of [nvidia/mit-b0](https://huggingface.co/nvidia/mit-b0) on the tontokoton/artery-ultrasound-siit dataset.
-It achieves the following results on the evaluation set:
-- eval_loss: 1.4971
-- eval_mean_iou: 0.1068
-- eval_mean_accuracy: 0.2327
-- eval_overall_accuracy: 0.8437
-- eval_accuracy_artery: nan
-- eval_accuracy_vein: 0.0062
-- eval_accuracy_nerve: 0.0022
-- eval_accuracy_muscle1: 0.0
-- eval_accuracy_muscle2: 0.9224
-- eval_accuracy_muscle3: nan
-- eval_accuracy_muscle4: nan
-- eval_accuracy_unknown: nan
-- eval_iou_artery: 0.0
-- eval_iou_vein: 0.0061
-- eval_iou_nerve: 0.0021
-- eval_iou_muscle1: 0.0
-- eval_iou_muscle2: 0.8458
-- eval_iou_muscle3: 0.0
-- eval_iou_muscle4: 0.0
-- eval_iou_unknown: 0.0
-- eval_runtime: 11.3293
-- eval_samples_per_second: 1.765
-- eval_steps_per_second: 0.883
-- epoch: 2.25
-- step: 90
 
 ## Model description
 
@@ -61,8 +35,8 @@ More information needed
 
 The following hyperparameters were used during training:
 - learning_rate: 1e-05
-- train_batch_size: 2
-- eval_batch_size: 2
+- train_batch_size: 24
+- eval_batch_size: 24
 - seed: 42
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: linear

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:27f821f4bfa88c37f9490f896931e5edbd3b3c4faf83ba954a6284d26dd4e764
+oid sha256:3766a4330ba2d7df21ccd03f8c51fa61b91624920249c6068c1c5ee30ed13ab9
 size 4091