Caraaaaa commited on
Commit
e4e390f
1 Parent(s): 61a7c17

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.0020
19
- - Mean Iou: 1.0
20
- - Mean Accuracy: 1.0
21
- - Overall Accuracy: 1.0
22
- - Per Category Iou: [1.0]
23
- - Per Category Accuracy: [1.0]
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 | Per Category Accuracy |
53
- |:-------------:|:-----:|:----:|:---------------:|:--------:|:-------------:|:----------------:|:----------------:|:---------------------:|
54
- | 0.0119 | 0.4 | 20 | 0.0266 | 1.0 | 1.0 | 1.0 | [1.0] | [1.0] |
55
- | 0.0049 | 0.8 | 40 | 0.0080 | 1.0 | 1.0 | 1.0 | [1.0] | [1.0] |
56
- | 0.0014 | 1.2 | 60 | 0.0038 | 1.0 | 1.0 | 1.0 | [1.0] | [1.0] |
57
- | 0.0043 | 1.6 | 80 | 0.0048 | 1.0 | 1.0 | 1.0 | [1.0] | [1.0] |
58
- | 0.0014 | 2.0 | 100 | 0.0032 | 1.0 | 1.0 | 1.0 | [1.0] | [1.0] |
59
- | 0.0009 | 2.4 | 120 | 0.0034 | 1.0 | 1.0 | 1.0 | [1.0] | [1.0] |
60
- | 0.0029 | 2.8 | 140 | 0.0027 | 1.0 | 1.0 | 1.0 | [1.0] | [1.0] |
61
- | 0.0007 | 3.2 | 160 | 0.0024 | 1.0 | 1.0 | 1.0 | [1.0] | [1.0] |
62
- | 0.0021 | 3.6 | 180 | 0.0025 | 1.0 | 1.0 | 1.0 | [1.0] | [1.0] |
63
- | 0.0014 | 4.0 | 200 | 0.0023 | 1.0 | 1.0 | 1.0 | [1.0] | [1.0] |
64
- | 0.0012 | 4.4 | 220 | 0.0021 | 1.0 | 1.0 | 1.0 | [1.0] | [1.0] |
65
- | 0.0027 | 4.8 | 240 | 0.0020 | 1.0 | 1.0 | 1.0 | [1.0] | [1.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:acb92fb212a1e0c4440ea2c202de44af9b0c2e1c82c6216b88fbbf730b431b3f
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:e5f4c4c2307306afe64b0e973c808358ce15b617d3e6f8f9f925be3afa7ab277
3
- size 40060
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9652d118d38e8be3015837ca90fb40c138c2a4393507820c5b01b9067fa2aa78
3
+ size 49144