Training in progress epoch 5

README.md

@@ -14,14 +14,14 @@ 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.1018
-- Train Dice Coef: 0.7226
-- Train Iou: 0.5858
-- Validation Loss: 0.0932
-- Validation Dice Coef: 0.7480
-- Validation Iou: 0.6051
+- Train Loss: 0.0930
+- Train Dice Coef: 0.7373
+- Train Iou: 0.6042
+- Validation Loss: 0.1618
+- Validation Dice Coef: 0.5048
+- Validation Iou: 0.3614
 - Train Lr: 1e-04
-- Epoch: 4
+- Epoch: 5
 
 ## Model description
 
@@ -52,6 +52,7 @@ The following hyperparameters were used during training:
 | 0.1295     | 0.6470          | 0.5014    | 0.1637          | 0.4628               | 0.3163         | 1e-04    | 2     |
 | 0.1109     | 0.6903          | 0.5511    | 0.1319          | 0.5634               | 0.4072         | 1e-04    | 3     |
 | 0.1018     | 0.7226          | 0.5858    | 0.0932          | 0.7480               | 0.6051         | 1e-04    | 4     |
+| 0.0930     | 0.7373          | 0.6042    | 0.1618          | 0.5048               | 0.3614         | 1e-04    | 5     |
 
 
 ### Framework versions

tf_model.h5

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:540471a010cc846c2072e5af506a5edbfae5a84d34ce1917f324ec136880bb9e
+oid sha256:6a36218fffa62f826622bd1205920216361b188cc09489510d8b91fc862eb040
 size 15133664