End of training

README.md

@@ -15,17 +15,17 @@ should probably proofread and complete it, then remove this comment. -->
 
 This model is a fine-tuned version of [nvidia/mit-b0](https://huggingface.co/nvidia/mit-b0) on an unknown dataset.
 It achieves the following results on the evaluation set:
-- eval_loss: 0.0019
+- eval_loss: 0.0079
 - eval_mean_iou: 1.0
 - eval_mean_accuracy: 1.0
 - eval_overall_accuracy: 1.0
 - eval_per_category_iou: [1.0]
 - eval_per_category_accuracy: [1.0]
-- eval_runtime: 59.0904
-- eval_samples_per_second: 0.846
-- eval_steps_per_second: 0.423
-- epoch: 2.0
-- step: 100
+- eval_runtime: 58.744
+- eval_samples_per_second: 0.851
+- eval_steps_per_second: 0.426
+- epoch: 1.6
+- step: 80
 
 ## Model description
 
@@ -50,7 +50,7 @@ The following hyperparameters were used during training:
 - seed: 42
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: linear
-- num_epochs: 50
+- num_epochs: 5
 
 ### Framework versions
 

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:bdc5e0374e637e27792e1b20d6e21a473b60463d3f63eb4cd57494a4d0c81c3c
+oid sha256:15e909c65639e9b3407db1b9ceaf3549d8b445b62ccd7dec4a76cbafb0d04979
 size 14883748

runs/Dec20_02-39-14_37b315fcf8c8/events.out.tfevents.1703039955.37b315fcf8c8.804.2

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:40f2c20cc2981ec29b8acef92fd326e544ecee3438d680e59637a134f48f81ef
-size 18693
+oid sha256:f5bd05d4cfca78704b4d8ede359532aacffd915f1a2f82f7c8e1d31e8beececd
+size 21773