Training in progress epoch 34

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.1922
-- Validation Loss: 0.6796
-- Validation Mean Iou: 0.3286
-- Validation Mean Accuracy: 0.4105
-- Validation Overall Accuracy: 0.8482
-- Validation Per Category Iou: [0.00000000e+00 7.40810462e-01 8.66118742e-01 4.99364852e-01
- 5.26878756e-01 3.72552936e-01            nan 4.74841300e-01
- 5.25079443e-01 1.14740050e-01 8.35714370e-01 6.38207346e-04
- 0.00000000e+00 0.00000000e+00 0.00000000e+00 5.74621701e-01
- 0.00000000e+00 0.00000000e+00 7.07495193e-01 1.41447126e-01
- 4.62213084e-01 3.40642698e-01 0.00000000e+00            nan
- 0.00000000e+00 3.89053890e-01 2.22898422e-01 0.00000000e+00
- 8.45321982e-01 7.94221222e-01 9.17872097e-01 1.76532216e-02
- 2.30290833e-01 2.42034008e-01 0.00000000e+00]
-- Validation Per Category Accuracy: [0.00000000e+00 8.16530181e-01 9.52002646e-01 6.63124748e-01
- 6.26037619e-01 4.92956588e-01            nan 6.47597208e-01
- 6.85001856e-01 1.36698809e-01 9.31706346e-01 7.06048482e-04
- 0.00000000e+00            nan 0.00000000e+00 6.94635901e-01
- 0.00000000e+00 0.00000000e+00 8.95976958e-01 1.87069749e-01
- 5.59829035e-01 4.10310044e-01 0.00000000e+00            nan
- 0.00000000e+00 4.87146698e-01 2.85578447e-01 0.00000000e+00
- 9.25510256e-01 9.23843473e-01 9.69757916e-01 4.94731348e-02
- 4.80154090e-01 3.14923397e-01 0.00000000e+00]
-- Epoch: 33
+- Train Loss: 0.1946
+- Validation Loss: 0.6672
+- Validation Mean Iou: 0.3339
+- Validation Mean Accuracy: 0.4209
+- Validation Overall Accuracy: 0.8513
+- Validation Per Category Iou: [0.         0.70907556 0.87023802 0.51805269 0.49452017 0.43417189
+        nan 0.4707293  0.53176677 0.07924453 0.81809353 0.00447818
+ 0.         0.         0.07079646 0.56036682 0.         0.
+ 0.71873076 0.12939172 0.4609741  0.40598735 0.                nan
+ 0.00292085 0.38228351 0.22329377 0.         0.85751928 0.83382942
+ 0.92111467 0.00420779 0.2179034  0.29863321 0.        ]
+- Validation Per Category Accuracy: [0.         0.75629783 0.9496271  0.64646194 0.78470005 0.53164143
+        nan 0.62385316 0.71429455 0.08996029 0.91607252 0.00643289
+ 0.                nan 0.07079646 0.77060234 0.         0.
+ 0.87272877 0.16646642 0.64641331 0.51623095 0.                nan
+ 0.00420521 0.43757258 0.28621236 0.         0.95711263 0.9030664
+ 0.96767724 0.00712818 0.47279356 0.37110309 0.        ]
+- Epoch: 34
 
 ## Model description
 
@@ -497,6 +491,17 @@ The following hyperparameters were used during training:
  0.00000000e+00 4.87146698e-01 2.85578447e-01 0.00000000e+00
  9.25510256e-01 9.23843473e-01 9.69757916e-01 4.94731348e-02
  4.80154090e-01 3.14923397e-01 0.00000000e+00] | 33    |
+| 0.1946     | 0.6672          | 0.3339              | 0.4209                   | 0.8513                      | [0.         0.70907556 0.87023802 0.51805269 0.49452017 0.43417189
+        nan 0.4707293  0.53176677 0.07924453 0.81809353 0.00447818
+ 0.         0.         0.07079646 0.56036682 0.         0.
+ 0.71873076 0.12939172 0.4609741  0.40598735 0.                nan
+ 0.00292085 0.38228351 0.22329377 0.         0.85751928 0.83382942
+ 0.92111467 0.00420779 0.2179034  0.29863321 0.        ]                                                                                                                                                        | [0.         0.75629783 0.9496271  0.64646194 0.78470005 0.53164143
+        nan 0.62385316 0.71429455 0.08996029 0.91607252 0.00643289
+ 0.                nan 0.07079646 0.77060234 0.         0.
+ 0.87272877 0.16646642 0.64641331 0.51623095 0.                nan
+ 0.00420521 0.43757258 0.28621236 0.         0.95711263 0.9030664
+ 0.96767724 0.00712818 0.47279356 0.37110309 0.        ]                                                                                                                                                         | 34    |
 
 
 ### Framework versions

tf_model.h5

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:156179fe75f1817c4eb85cec9f5dd1f797648e504da1b56aa1338a4e600cc8dc
+oid sha256:53d739e4a7c93bfe786f9d21d0f819fadd013b06e9714931bc1c8bea63f0db21
 size 15167588