Rolv-Arild commited on
Commit
cbf2f24
·
verified ·
1 Parent(s): 72462f5

Model save

Browse files
Files changed (2) hide show
  1. README.md +42 -38
  2. model.safetensors +1 -1
README.md CHANGED
@@ -3,31 +3,29 @@ library_name: transformers
3
  license: other
4
  base_model: nvidia/mit-b0
5
  tags:
6
- - image-segmentation
7
- - vision
8
  - generated_from_trainer
9
  datasets:
10
  - generator
11
  model-index:
12
- - name: autocrop-test
13
  results: []
14
  ---
15
 
16
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
17
  should probably proofread and complete it, then remove this comment. -->
18
 
19
- # autocrop-test
20
 
21
- This model is a fine-tuned version of [nvidia/mit-b0](https://huggingface.co/nvidia/mit-b0) on the /mnt/disk1/autocrop-data/datasets/tekst/ dataset.
22
  It achieves the following results on the evaluation set:
23
- - Loss: 0.0196
24
- - Mean Iou: 0.4964
25
- - Mean Accuracy: 0.9928
26
- - Overall Accuracy: 0.9928
27
  - Accuracy Background: nan
28
- - Accuracy Crop: 0.9928
29
  - Iou Background: 0.0
30
- - Iou Crop: 0.9928
31
 
32
  ## Model description
33
 
@@ -58,33 +56,39 @@ The following hyperparameters were used during training:
58
 
59
  ### Training results
60
 
61
- | Training Loss | Epoch | Step | Validation Loss | Mean Iou | Mean Accuracy | Overall Accuracy | Accuracy Background | Accuracy Crop | Iou Background | Iou Crop |
62
- |:-------------:|:-----:|:-----:|:---------------:|:--------:|:-------------:|:----------------:|:-------------------:|:-------------:|:--------------:|:--------:|
63
- | 0.4334 | 1.0 | 625 | 0.1014 | 0.4884 | 0.9768 | 0.9768 | nan | 0.9768 | 0.0 | 0.9768 |
64
- | 0.1136 | 2.0 | 1250 | 0.0503 | 0.4942 | 0.9883 | 0.9883 | nan | 0.9883 | 0.0 | 0.9883 |
65
- | 0.0626 | 3.0 | 1875 | 0.0365 | 0.4952 | 0.9903 | 0.9903 | nan | 0.9903 | 0.0 | 0.9903 |
66
- | 0.0376 | 4.0 | 2500 | 0.0304 | 0.4957 | 0.9913 | 0.9913 | nan | 0.9913 | 0.0 | 0.9913 |
67
- | 0.0331 | 5.0 | 3125 | 0.0277 | 0.4945 | 0.9890 | 0.9890 | nan | 0.9890 | 0.0 | 0.9890 |
68
- | 0.0312 | 6.0 | 3750 | 0.0258 | 0.4961 | 0.9922 | 0.9922 | nan | 0.9922 | 0.0 | 0.9922 |
69
- | 0.0290 | 7.0 | 4375 | 0.0252 | 0.4970 | 0.9941 | 0.9941 | nan | 0.9941 | 0.0 | 0.9941 |
70
- | 0.0272 | 8.0 | 5000 | 0.0247 | 0.4950 | 0.9900 | 0.9900 | nan | 0.9900 | 0.0 | 0.9900 |
71
- | 0.0264 | 9.0 | 5625 | 0.0232 | 0.4963 | 0.9925 | 0.9925 | nan | 0.9925 | 0.0 | 0.9925 |
72
- | 0.0260 | 10.0 | 6250 | 0.0230 | 0.4964 | 0.9927 | 0.9927 | nan | 0.9927 | 0.0 | 0.9927 |
73
- | 0.0257 | 11.0 | 6875 | 0.0223 | 0.4969 | 0.9937 | 0.9937 | nan | 0.9937 | 0.0 | 0.9937 |
74
- | 0.0244 | 12.0 | 7500 | 0.0217 | 0.4966 | 0.9932 | 0.9932 | nan | 0.9932 | 0.0 | 0.9932 |
75
- | 0.0240 | 13.0 | 8125 | 0.0223 | 0.4960 | 0.9920 | 0.9920 | nan | 0.9920 | 0.0 | 0.9920 |
76
- | 0.0230 | 14.0 | 8750 | 0.0220 | 0.4972 | 0.9943 | 0.9943 | nan | 0.9943 | 0.0 | 0.9943 |
77
- | 0.0230 | 15.0 | 9375 | 0.0213 | 0.4963 | 0.9926 | 0.9926 | nan | 0.9926 | 0.0 | 0.9926 |
78
- | 0.0228 | 16.0 | 10000 | 0.0208 | 0.4964 | 0.9928 | 0.9928 | nan | 0.9928 | 0.0 | 0.9928 |
79
- | 0.0220 | 17.0 | 10625 | 0.0206 | 0.4965 | 0.9930 | 0.9930 | nan | 0.9930 | 0.0 | 0.9930 |
80
- | 0.0217 | 18.0 | 11250 | 0.0205 | 0.4960 | 0.9921 | 0.9921 | nan | 0.9921 | 0.0 | 0.9921 |
81
- | 0.0212 | 19.0 | 11875 | 0.0207 | 0.4958 | 0.9915 | 0.9915 | nan | 0.9915 | 0.0 | 0.9915 |
82
- | 0.0209 | 20.0 | 12500 | 0.0204 | 0.4973 | 0.9946 | 0.9946 | nan | 0.9946 | 0.0 | 0.9946 |
83
- | 0.0203 | 21.0 | 13125 | 0.0198 | 0.4969 | 0.9937 | 0.9937 | nan | 0.9937 | 0.0 | 0.9937 |
84
- | 0.0202 | 22.0 | 13750 | 0.0196 | 0.4964 | 0.9928 | 0.9928 | nan | 0.9928 | 0.0 | 0.9928 |
85
- | 0.0202 | 23.0 | 14375 | 0.0203 | 0.4971 | 0.9942 | 0.9942 | nan | 0.9942 | 0.0 | 0.9942 |
86
- | 0.0198 | 24.0 | 15000 | 0.0196 | 0.4966 | 0.9932 | 0.9932 | nan | 0.9932 | 0.0 | 0.9932 |
87
- | 0.0194 | 25.0 | 15625 | 0.0197 | 0.4966 | 0.9932 | 0.9932 | nan | 0.9932 | 0.0 | 0.9932 |
 
 
 
 
 
 
88
 
89
 
90
  ### Framework versions
 
3
  license: other
4
  base_model: nvidia/mit-b0
5
  tags:
 
 
6
  - generated_from_trainer
7
  datasets:
8
  - generator
9
  model-index:
10
+ - name: autocrop-bilder
11
  results: []
12
  ---
13
 
14
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
15
  should probably proofread and complete it, then remove this comment. -->
16
 
17
+ # autocrop-bilder
18
 
19
+ This model is a fine-tuned version of [nvidia/mit-b0](https://huggingface.co/nvidia/mit-b0) on the generator dataset.
20
  It achieves the following results on the evaluation set:
21
+ - Loss: 0.1091
22
+ - Mean Iou: 0.4907
23
+ - Mean Accuracy: 0.9814
24
+ - Overall Accuracy: 0.9814
25
  - Accuracy Background: nan
26
+ - Accuracy Crop: 0.9814
27
  - Iou Background: 0.0
28
+ - Iou Crop: 0.9814
29
 
30
  ## Model description
31
 
 
56
 
57
  ### Training results
58
 
59
+ | Training Loss | Epoch | Step | Validation Loss | Mean Iou | Mean Accuracy | Overall Accuracy | Accuracy Background | Accuracy Crop | Iou Background | Iou Crop |
60
+ |:-------------:|:-----:|:----:|:---------------:|:--------:|:-------------:|:----------------:|:-------------------:|:-------------:|:--------------:|:--------:|
61
+ | No log | 1.0 | 7 | 0.6694 | 0.3399 | 0.6798 | 0.6798 | nan | 0.6798 | 0.0 | 0.6798 |
62
+ | No log | 2.0 | 14 | 0.5930 | 0.4623 | 0.9246 | 0.9246 | nan | 0.9246 | 0.0 | 0.9246 |
63
+ | No log | 3.0 | 21 | 0.4610 | 0.4706 | 0.9412 | 0.9412 | nan | 0.9412 | 0.0 | 0.9412 |
64
+ | No log | 4.0 | 28 | 0.3075 | 0.4705 | 0.9411 | 0.9411 | nan | 0.9411 | 0.0 | 0.9411 |
65
+ | No log | 5.0 | 35 | 0.2037 | 0.4709 | 0.9417 | 0.9417 | nan | 0.9417 | 0.0 | 0.9417 |
66
+ | No log | 6.0 | 42 | 0.1668 | 0.4662 | 0.9324 | 0.9324 | nan | 0.9324 | 0.0 | 0.9324 |
67
+ | No log | 7.0 | 49 | 0.1421 | 0.4752 | 0.9503 | 0.9503 | nan | 0.9503 | 0.0 | 0.9503 |
68
+ | No log | 8.0 | 56 | 0.1382 | 0.4773 | 0.9547 | 0.9547 | nan | 0.9547 | 0.0 | 0.9547 |
69
+ | No log | 9.0 | 63 | 0.1588 | 0.4737 | 0.9473 | 0.9473 | nan | 0.9473 | 0.0 | 0.9473 |
70
+ | No log | 10.0 | 70 | 0.1317 | 0.4845 | 0.9690 | 0.9690 | nan | 0.9690 | 0.0 | 0.9690 |
71
+ | No log | 11.0 | 77 | 0.1307 | 0.4836 | 0.9671 | 0.9671 | nan | 0.9671 | 0.0 | 0.9671 |
72
+ | No log | 12.0 | 84 | 0.1328 | 0.4899 | 0.9798 | 0.9798 | nan | 0.9798 | 0.0 | 0.9798 |
73
+ | No log | 13.0 | 91 | 0.1265 | 0.4863 | 0.9725 | 0.9725 | nan | 0.9725 | 0.0 | 0.9725 |
74
+ | No log | 14.0 | 98 | 0.1283 | 0.4886 | 0.9772 | 0.9772 | nan | 0.9772 | 0.0 | 0.9772 |
75
+ | No log | 15.0 | 105 | 0.1286 | 0.4887 | 0.9775 | 0.9775 | nan | 0.9775 | 0.0 | 0.9775 |
76
+ | No log | 16.0 | 112 | 0.1235 | 0.4894 | 0.9788 | 0.9788 | nan | 0.9788 | 0.0 | 0.9788 |
77
+ | No log | 17.0 | 119 | 0.1213 | 0.4898 | 0.9795 | 0.9795 | nan | 0.9795 | 0.0 | 0.9795 |
78
+ | No log | 18.0 | 126 | 0.1223 | 0.4910 | 0.9821 | 0.9821 | nan | 0.9821 | 0.0 | 0.9821 |
79
+ | No log | 19.0 | 133 | 0.1179 | 0.4882 | 0.9763 | 0.9763 | nan | 0.9763 | 0.0 | 0.9763 |
80
+ | No log | 20.0 | 140 | 0.1169 | 0.4914 | 0.9829 | 0.9829 | nan | 0.9829 | 0.0 | 0.9829 |
81
+ | No log | 21.0 | 147 | 0.1153 | 0.4908 | 0.9816 | 0.9816 | nan | 0.9816 | 0.0 | 0.9816 |
82
+ | No log | 22.0 | 154 | 0.1155 | 0.4902 | 0.9804 | 0.9804 | nan | 0.9804 | 0.0 | 0.9804 |
83
+ | No log | 23.0 | 161 | 0.1143 | 0.4919 | 0.9839 | 0.9839 | nan | 0.9839 | 0.0 | 0.9839 |
84
+ | No log | 24.0 | 168 | 0.1115 | 0.4913 | 0.9825 | 0.9825 | nan | 0.9825 | 0.0 | 0.9825 |
85
+ | No log | 25.0 | 175 | 0.1113 | 0.4920 | 0.9841 | 0.9841 | nan | 0.9841 | 0.0 | 0.9841 |
86
+ | No log | 26.0 | 182 | 0.1106 | 0.4923 | 0.9846 | 0.9846 | nan | 0.9846 | 0.0 | 0.9846 |
87
+ | No log | 27.0 | 189 | 0.1097 | 0.4907 | 0.9814 | 0.9814 | nan | 0.9814 | 0.0 | 0.9814 |
88
+ | No log | 28.0 | 196 | 0.1087 | 0.4908 | 0.9816 | 0.9816 | nan | 0.9816 | 0.0 | 0.9816 |
89
+ | No log | 29.0 | 203 | 0.1094 | 0.4920 | 0.9840 | 0.9840 | nan | 0.9840 | 0.0 | 0.9840 |
90
+ | No log | 30.0 | 210 | 0.1093 | 0.4911 | 0.9821 | 0.9821 | nan | 0.9821 | 0.0 | 0.9821 |
91
+ | No log | 31.0 | 217 | 0.1091 | 0.4907 | 0.9814 | 0.9814 | nan | 0.9814 | 0.0 | 0.9814 |
92
 
93
 
94
  ### Framework versions
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:9f7d7621281615bf843731f9ee939e86179490ea7c8e159db82dadf91ab1ed2e
3
  size 14884776
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a9342f22c1483405051f2262fb32d379bd8466a36900131a3efd5e9618a06b74
3
  size 14884776