Training in progress epoch 2

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.1354
-- Train Dice Coef: 0.6853
-- Train Iou: 0.5329
-- Validation Loss: 0.0855
-- Validation Dice Coef: 0.7853
-- Validation Iou: 0.6544
+- Train Loss: 0.1105
+- Train Dice Coef: 0.7364
+- Train Iou: 0.5924
+- Validation Loss: 0.0737
+- Validation Dice Coef: 0.8147
+- Validation Iou: 0.6916
 - Train Lr: 1e-04
-- Epoch: 1
+- Epoch: 2
 
 ## Model description
 
@@ -49,6 +49,7 @@ The following hyperparameters were used during training:
 |:----------:|:---------------:|:---------:|:---------------:|:--------------------:|:--------------:|:--------:|:-----:|
 | 0.2282     | 0.5657          | 0.4102    | 0.1322          | 0.6524               | 0.4967         | 1e-04    | 0     |
 | 0.1354     | 0.6853          | 0.5329    | 0.0855          | 0.7853               | 0.6544         | 1e-04    | 1     |
+| 0.1105     | 0.7364          | 0.5924    | 0.0737          | 0.8147               | 0.6916         | 1e-04    | 2     |
 
 
 ### Framework versions

tf_model.h5

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:62849af95357753001f8f534504997c08c08d1e96a2b62aa8c4b463b12bbaa8c
+oid sha256:55e97ca7fd8750f298d966acb72beaf2162f8bb049fbefca17e5e2dfe8af70e0
 size 15133664