End of training
Browse files
README.md
CHANGED
@@ -15,12 +15,12 @@ should probably proofread and complete it, then remove this comment. -->
|
|
15 |
|
16 |
This model is a fine-tuned version of [nvidia/mit-b0](https://huggingface.co/nvidia/mit-b0) on an unknown dataset.
|
17 |
It achieves the following results on the evaluation set:
|
18 |
-
- Loss: 0.
|
19 |
-
- Mean Iou:
|
20 |
-
- Mean Accuracy:
|
21 |
-
- Overall Accuracy:
|
22 |
-
- Per Category Iou: [
|
23 |
-
- Per Category Accuracy: [
|
24 |
|
25 |
## Model description
|
26 |
|
@@ -49,20 +49,20 @@ The following hyperparameters were used during training:
|
|
49 |
|
50 |
### Training results
|
51 |
|
52 |
-
| Training Loss | Epoch | Step | Validation Loss | Mean Iou | Mean Accuracy | Overall Accuracy | Per Category Iou
|
53 |
-
|
54 |
-
| 0.
|
55 |
-
| 0.
|
56 |
-
| 0.
|
57 |
-
| 0.
|
58 |
-
| 0.
|
59 |
-
| 0.
|
60 |
-
| 0.
|
61 |
-
| 0.
|
62 |
-
| 0.
|
63 |
-
| 0.
|
64 |
-
| 0.
|
65 |
-
| 0.
|
66 |
|
67 |
|
68 |
### Framework versions
|
|
|
15 |
|
16 |
This model is a fine-tuned version of [nvidia/mit-b0](https://huggingface.co/nvidia/mit-b0) on an unknown dataset.
|
17 |
It achieves the following results on the evaluation set:
|
18 |
+
- Loss: 0.2273
|
19 |
+
- Mean Iou: 0.7888
|
20 |
+
- Mean Accuracy: 0.8815
|
21 |
+
- Overall Accuracy: 0.9466
|
22 |
+
- Per Category Iou: [0.9411436688097194, 0.6365339140286779]
|
23 |
+
- Per Category Accuracy: [0.9666298272627839, 0.7963078307432842]
|
24 |
|
25 |
## Model description
|
26 |
|
|
|
49 |
|
50 |
### Training results
|
51 |
|
52 |
+
| Training Loss | Epoch | Step | Validation Loss | Mean Iou | Mean Accuracy | Overall Accuracy | Per Category Iou | Per Category Accuracy |
|
53 |
+
|:-------------:|:-----:|:----:|:---------------:|:--------:|:-------------:|:----------------:|:-----------------------------------------:|:----------------------------------------:|
|
54 |
+
| 0.6854 | 0.4 | 20 | 0.6497 | 0.5097 | 0.8417 | 0.7443 | [0.7115032459977992, 0.3077993102868545] | [0.7144421527142827, 0.9689304620217858] |
|
55 |
+
| 0.5904 | 0.8 | 40 | 0.5143 | 0.5981 | 0.8004 | 0.8478 | [0.8333935175312267, 0.3627987944393932] | [0.862384168080342, 0.7383421777739057] |
|
56 |
+
| 0.4687 | 1.2 | 60 | 0.4546 | 0.6050 | 0.8161 | 0.8493 | [0.8342992363445556, 0.37569323794278087] | [0.8595097956337959, 0.7727067629087438] |
|
57 |
+
| 0.424 | 1.6 | 80 | 0.3844 | 0.6584 | 0.8272 | 0.8881 | [0.8773645633114495, 0.43953238690068724] | [0.9067963946134603, 0.7476733768251818] |
|
58 |
+
| 0.3485 | 2.0 | 100 | 0.3535 | 0.6942 | 0.8808 | 0.8995 | [0.8882922581800861, 0.5000111990627714] | [0.9052579907882474, 0.8563377644392374] |
|
59 |
+
| 0.3402 | 2.4 | 120 | 0.3424 | 0.7021 | 0.9165 | 0.8977 | [0.8850157203428243, 0.5191490567411473] | [0.891946363487535, 0.9410955056545037] |
|
60 |
+
| 0.3037 | 2.8 | 140 | 0.3071 | 0.7357 | 0.9150 | 0.9173 | [0.9073510172037412, 0.5639900768760655] | [0.9179667207588744, 0.9119966243321158] |
|
61 |
+
| 0.2669 | 3.2 | 160 | 0.2528 | 0.7659 | 0.8552 | 0.9410 | [0.9353899879693552, 0.5964963563086936] | [0.9673518346317019, 0.7429811204460667] |
|
62 |
+
| 0.225 | 3.6 | 180 | 0.2523 | 0.7695 | 0.8894 | 0.9375 | [0.9308386361574114, 0.6082593419593063] | [0.9523221012048093, 0.826397566814428] |
|
63 |
+
| 0.2703 | 4.0 | 200 | 0.2430 | 0.7812 | 0.8943 | 0.9419 | [0.9355827639904635, 0.6268385945303665] | [0.9564666637969618, 0.8320982324555644] |
|
64 |
+
| 0.2607 | 4.4 | 220 | 0.2352 | 0.7875 | 0.8918 | 0.9448 | [0.9389026387973957, 0.6361696692483538] | [0.9610553391463873, 0.822527121722995] |
|
65 |
+
| 0.2314 | 4.8 | 240 | 0.2273 | 0.7888 | 0.8815 | 0.9466 | [0.9411436688097194, 0.6365339140286779] | [0.9666298272627839, 0.7963078307432842] |
|
66 |
|
67 |
|
68 |
### Framework versions
|
model.safetensors
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
size 14884776
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:9ea7560099739906b70aea56b5853de56e0cdd173f4569562ba1ae4c318b43c6
|
3 |
size 14884776
|
runs/Dec20_03-07-08_37b315fcf8c8/events.out.tfevents.1703041629.37b315fcf8c8.12167.0
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
-
size
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:9652d118d38e8be3015837ca90fb40c138c2a4393507820c5b01b9067fa2aa78
|
3 |
+
size 49144
|