Training in progress epoch 18

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.2833
-- Validation Loss: 0.6446
-- Validation Mean Iou: 0.3261
-- Validation Mean Accuracy: 0.3949
-- Validation Overall Accuracy: 0.8447
-- Validation Per Category Iou: [0.00000000e+00 7.08995850e-01 8.49245624e-01 5.02729624e-01
- 5.44178924e-01 2.80874020e-01            nan 4.09531940e-01
- 4.68642701e-01 5.72334850e-02 7.86049035e-01 4.15253651e-04
- 0.00000000e+00            nan 0.00000000e+00 5.20689462e-01
- 0.00000000e+00 0.00000000e+00 7.19140232e-01 1.09126891e-01
- 4.66887817e-01 3.77114972e-01 0.00000000e+00            nan
- 0.00000000e+00 3.57675941e-01 1.49472418e-01 0.00000000e+00
- 8.48995042e-01 8.23287514e-01 9.17986999e-01 4.86177781e-03
- 2.08704959e-01 3.22665861e-01 0.00000000e+00]
-- Validation Per Category Accuracy: [0.00000000e+00 8.00869189e-01 9.55886409e-01 6.05400396e-01
- 6.79327160e-01 3.35582853e-01            nan 5.64117225e-01
- 7.19032298e-01 7.06182643e-02 9.12828794e-01 4.70698988e-04
- 0.00000000e+00            nan 0.00000000e+00 6.96671162e-01
- 0.00000000e+00 0.00000000e+00 8.85781041e-01 1.13699831e-01
- 6.05324120e-01 4.37527674e-01 0.00000000e+00            nan
- 0.00000000e+00 4.41345654e-01 1.73312203e-01 0.00000000e+00
- 9.54319708e-01 8.94089174e-01 9.60961557e-01 6.44308877e-03
- 4.22542478e-01 3.99729132e-01 0.00000000e+00]
-- Epoch: 17
+- Train Loss: 0.2777
+- Validation Loss: 0.5923
+- Validation Mean Iou: 0.3311
+- Validation Mean Accuracy: 0.4005
+- Validation Overall Accuracy: 0.8496
+- Validation Per Category Iou: [0.         0.74552742 0.85847838 0.50817039 0.54300979 0.31118625
+        nan 0.42604196 0.48432857 0.         0.80689375 0.
+ 0.                nan 0.         0.53701465 0.         0.
+ 0.72095961 0.1181785  0.44646925 0.38422934 0.                nan
+ 0.         0.36931298 0.14420905 0.         0.85509203 0.82052953
+ 0.91364053 0.00100348 0.23289685 0.36817268 0.        ]
+- Validation Per Category Accuracy: [0.         0.84028626 0.9434955  0.64877174 0.63691529 0.41306725
+        nan 0.60885149 0.7505436  0.         0.92003686 0.
+ 0.                nan 0.         0.6933623  0.         0.
+ 0.87623974 0.12084362 0.60794245 0.40909699 0.                nan
+ 0.         0.43594135 0.17001585 0.         0.95100912 0.93487828
+ 0.9639534  0.00228363 0.4196533  0.46863408 0.        ]
+- Epoch: 18
 
 ## Model description
 
@@ -285,6 +279,17 @@ The following hyperparameters were used during training:
  0.00000000e+00 4.41345654e-01 1.73312203e-01 0.00000000e+00
  9.54319708e-01 8.94089174e-01 9.60961557e-01 6.44308877e-03
  4.22542478e-01 3.99729132e-01 0.00000000e+00] | 17    |
+| 0.2777     | 0.5923          | 0.3311              | 0.4005                   | 0.8496                      | [0.         0.74552742 0.85847838 0.50817039 0.54300979 0.31118625
+        nan 0.42604196 0.48432857 0.         0.80689375 0.
+ 0.                nan 0.         0.53701465 0.         0.
+ 0.72095961 0.1181785  0.44646925 0.38422934 0.                nan
+ 0.         0.36931298 0.14420905 0.         0.85509203 0.82052953
+ 0.91364053 0.00100348 0.23289685 0.36817268 0.        ]                                                                                                                                                                | [0.         0.84028626 0.9434955  0.64877174 0.63691529 0.41306725
+        nan 0.60885149 0.7505436  0.         0.92003686 0.
+ 0.                nan 0.         0.6933623  0.         0.
+ 0.87623974 0.12084362 0.60794245 0.40909699 0.                nan
+ 0.         0.43594135 0.17001585 0.         0.95100912 0.93487828
+ 0.9639534  0.00228363 0.4196533  0.46863408 0.        ]                                                                                                                                                                | 18    |
 
 
 ### Framework versions

tf_model.h5

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:f4f04e5cd78114c77130df86a4314b9347c65685acd01b9c3e0bbbac8cb576f4
+oid sha256:d79159597af382f22163832cb4e14ff516576afe8d4f31ff9a850ffaa923025e
 size 15167588