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-b4](https://huggingface.co/nvidia/mit-b4) on the None dataset.
 It achieves the following results on the evaluation set:
-- eval_loss: 0.1324
-- eval_mean_iou: 0.6195
-- eval_mean_accuracy: 0.6975
-- eval_overall_accuracy: 0.9546
-- eval_per_category_iou: [0.9539834955057686, 0.5323742381722192, 0.6195781259318675, 0.37206494047766914]
-- eval_per_category_accuracy: [0.9837435927526105, 0.655434290717248, 0.7038678684651577, 0.44679832019118265]
-- eval_runtime: 1163.7795
-- eval_samples_per_second: 0.711
-- eval_steps_per_second: 0.178
-- epoch: 10.0
-- step: 9300
+- eval_loss: 0.1252
+- eval_mean_iou: 0.6127
+- eval_mean_accuracy: 0.6824
+- eval_overall_accuracy: 0.9576
+- eval_per_category_iou: [0.9565401111001023, 0.5056775417338342, 0.6199915247210127, 0.3684174133417381]
+- eval_per_category_accuracy: [0.9868643557839373, 0.6055775784407109, 0.6926147942226376, 0.44438043699630503]
+- eval_runtime: 584.3554
+- eval_samples_per_second: 1.415
+- eval_steps_per_second: 0.354
+- epoch: 18.0
+- step: 16740
 
 ## Model description
 

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:66ce633ed9428f254d87ea5da2b1478fe676d003b3604010d32b24452913e3d0
+oid sha256:05043ff0f5130061765177b42746f8373e555b499fe8104e0065155102a91cd0
 size 256313237