Training in progress epoch 23

README.md

@@ -14,30 +14,24 @@ 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:
-- Train Loss: 0.2332
-- Validation Loss: 0.6204
-- Validation Mean Iou: 0.3332
-- Validation Mean Accuracy: 0.4059
-- Validation Overall Accuracy: 0.8483
-- Validation Per Category Iou: [0.00000000e+00 7.35393964e-01 8.60690644e-01 5.19021773e-01
- 5.30969013e-01 3.58918972e-01            nan 4.20497772e-01
- 4.95716847e-01 1.37272913e-02 8.19133033e-01 5.46224225e-04
- 0.00000000e+00            nan 0.00000000e+00 5.68465477e-01
- 0.00000000e+00 0.00000000e+00 7.20011345e-01 1.42622295e-01
- 4.17365004e-01 3.82529649e-01 0.00000000e+00            nan
- 6.70870790e-05 3.60538230e-01 1.69147635e-01 0.00000000e+00
- 8.55389638e-01 8.18918753e-01 9.17882158e-01 1.64044825e-02
- 2.01049360e-01 3.37421484e-01 0.00000000e+00]
-- Validation Per Category Accuracy: [0.00000000e+00 8.46018595e-01 9.32405953e-01 6.36213716e-01
- 6.56514609e-01 5.05878965e-01            nan 5.80474598e-01
- 7.45257836e-01 1.52580828e-02 9.27215688e-01 6.27598651e-04
- 0.00000000e+00            nan 0.00000000e+00 7.09744281e-01
- 0.00000000e+00 0.00000000e+00 8.72681917e-01 1.56351474e-01
- 5.64231108e-01 4.44759880e-01 0.00000000e+00            nan
- 1.40173816e-04 4.49791012e-01 2.10332805e-01 0.00000000e+00
- 9.49627606e-01 9.31113143e-01 9.68909254e-01 3.53635859e-02
- 4.13840545e-01 4.36765484e-01 0.00000000e+00]
-- Epoch: 22
+- Train Loss: 0.2324
+- Validation Loss: 0.6074
+- Validation Mean Iou: 0.3379
+- Validation Mean Accuracy: 0.4106
+- Validation Overall Accuracy: 0.8528
+- Validation Per Category Iou: [0.         0.72434007 0.87301941 0.51447452 0.65357748 0.31026479
+        nan 0.44044211 0.50481839 0.03262782 0.81648207 0.
+ 0.                nan 0.         0.57028611 0.         0.
+ 0.71488251 0.14924904 0.45551669 0.37382533 0.                nan
+ 0.         0.39635836 0.179016   0.         0.85971264 0.81216021
+ 0.92138172 0.01167807 0.1977878  0.29963924 0.        ]
+- Validation Per Category Accuracy: [0.         0.78602724 0.94880444 0.62217131 0.89442006 0.42531888
+        nan 0.59441554 0.64546467 0.03471356 0.9292238  0.
+ 0.                nan 0.         0.74615423 0.         0.
+ 0.88791835 0.16198532 0.66445513 0.45227575 0.                nan
+ 0.         0.4809923  0.21058637 0.         0.93914912 0.91867471
+ 0.96219496 0.02304832 0.45697186 0.35359374 0.        ]
+- Epoch: 23
 
 ## Model description
 
@@ -346,6 +340,17 @@ The following hyperparameters were used during training:
  1.40173816e-04 4.49791012e-01 2.10332805e-01 0.00000000e+00
  9.49627606e-01 9.31113143e-01 9.68909254e-01 3.53635859e-02
  4.13840545e-01 4.36765484e-01 0.00000000e+00] | 22    |
+| 0.2324     | 0.6074          | 0.3379              | 0.4106                   | 0.8528                      | [0.         0.72434007 0.87301941 0.51447452 0.65357748 0.31026479
+        nan 0.44044211 0.50481839 0.03262782 0.81648207 0.
+ 0.                nan 0.         0.57028611 0.         0.
+ 0.71488251 0.14924904 0.45551669 0.37382533 0.                nan
+ 0.         0.39635836 0.179016   0.         0.85971264 0.81216021
+ 0.92138172 0.01167807 0.1977878  0.29963924 0.        ]                                                                                                                                                                | [0.         0.78602724 0.94880444 0.62217131 0.89442006 0.42531888
+        nan 0.59441554 0.64546467 0.03471356 0.9292238  0.
+ 0.                nan 0.         0.74615423 0.         0.
+ 0.88791835 0.16198532 0.66445513 0.45227575 0.                nan
+ 0.         0.4809923  0.21058637 0.         0.93914912 0.91867471
+ 0.96219496 0.02304832 0.45697186 0.35359374 0.        ]                                                                                                                                                                | 23    |
 
 
 ### Framework versions

tf_model.h5

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:897c1a48620ba448828014cb6d4ead5d80a40480bdf98d35b15feba04bdc316f
+oid sha256:7dc13569d93cc9a0ad1ac81b3cddf08b59d47fa3316a9d459517803be12f3d96
 size 15167588