Training in progress epoch 4

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.0432
-- Train Dice Coef: 0.6204
-- Train Iou: 0.4699
-- Validation Loss: 0.1085
-- Validation Dice Coef: 0.3757
-- Validation Iou: 0.2479
+- Train Loss: 0.0413
+- Train Dice Coef: 0.6306
+- Train Iou: 0.4831
+- Validation Loss: 0.0771
+- Validation Dice Coef: 0.5239
+- Validation Iou: 0.3646
 - Train Lr: 1e-04
-- Epoch: 3
+- Epoch: 4
 
 ## Model description
 
@@ -51,6 +51,7 @@ The following hyperparameters were used during training:
 | 0.0610     | 0.5251          | 0.3760    | 0.1773          | 0.1542               | 0.0892         | 1e-04    | 1     |
 | 0.0500     | 0.5831          | 0.4322    | 0.0806          | 0.5067               | 0.3659         | 1e-04    | 2     |
 | 0.0432     | 0.6204          | 0.4699    | 0.1085          | 0.3757               | 0.2479         | 1e-04    | 3     |
+| 0.0413     | 0.6306          | 0.4831    | 0.0771          | 0.5239               | 0.3646         | 1e-04    | 4     |
 
 
 ### Framework versions

tf_model.h5

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:cdec0b0e5d5f9e6638bbae56a4daa75f850ed4186463a30160c8a5a40ffbad6f
+oid sha256:ee0dac8712cdbf1d558cf855e184492adfb9da74e11630ca54c0ece21d1ef6fe
 size 15133664