Training in progress epoch 25

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.0510
-- Train Dice Coef: 0.8634
-- Train Iou: 0.7624
-- Validation Loss: 0.0501
-- Validation Dice Coef: 0.8814
-- Validation Iou: 0.7899
-- Train Lr: 1e-04
-- Epoch: 24
+- Train Loss: 0.0485
+- Train Dice Coef: 0.8703
+- Train Iou: 0.7728
+- Validation Loss: 0.0439
+- Validation Dice Coef: 0.8892
+- Validation Iou: 0.8018
+- Train Lr: 5e-06
+- Epoch: 25
 
 ## Model description
 
@@ -72,6 +72,7 @@ The following hyperparameters were used during training:
 | 0.0518     | 0.8630          | 0.7615    | 0.0487          | 0.8836               | 0.7930         | 1e-04    | 22    |
 | 0.0521     | 0.8616          | 0.7595    | 0.0483          | 0.8822               | 0.7908         | 1e-04    | 23    |
 | 0.0510     | 0.8634          | 0.7624    | 0.0501          | 0.8814               | 0.7899         | 1e-04    | 24    |
+| 0.0485     | 0.8703          | 0.7728    | 0.0439          | 0.8892               | 0.8018         | 5e-06    | 25    |
 
 
 ### Framework versions

tf_model.h5

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:65eb3e10113f665420e6bea659be1ebf27ded61a603f9d97a3b86d0d1ba861f3
+oid sha256:85d4c3460d68502a3a453cef3592dbbff63b70dc91d7852df100061c2564da56
 size 15133664