Commit
·
2b4cb5c
1
Parent(s):
60c4cc9
Training in progress epoch 18
Browse files- README.md +6 -5
- tf_model.h5 +1 -1
README.md
CHANGED
|
@@ -14,14 +14,14 @@ probably proofread and complete it, then remove this comment. -->
|
|
| 14 |
|
| 15 |
This model is a fine-tuned version of [nvidia/mit-b0](https://huggingface.co/nvidia/mit-b0) on an unknown dataset.
|
| 16 |
It achieves the following results on the evaluation set:
|
| 17 |
-
- Train Loss: 0.
|
| 18 |
-
- Train Dice Coef: 0.
|
| 19 |
-
- Train Iou: 0.
|
| 20 |
-
- Validation Loss: 0.
|
| 21 |
- Validation Dice Coef: 0.8838
|
| 22 |
- Validation Iou: 0.7927
|
| 23 |
- Train Lr: 5e-06
|
| 24 |
-
- Epoch:
|
| 25 |
|
| 26 |
## Model description
|
| 27 |
|
|
@@ -65,6 +65,7 @@ The following hyperparameters were used during training:
|
|
| 65 |
| 0.0595 | 0.8160 | 0.7064 | 0.0475 | 0.8810 | 0.7883 | 5e-06 | 15 |
|
| 66 |
| 0.0589 | 0.8197 | 0.7115 | 0.0460 | 0.8835 | 0.7922 | 5e-06 | 16 |
|
| 67 |
| 0.0576 | 0.8214 | 0.7134 | 0.0459 | 0.8838 | 0.7927 | 5e-06 | 17 |
|
|
|
|
| 68 |
|
| 69 |
|
| 70 |
### Framework versions
|
|
|
|
| 14 |
|
| 15 |
This model is a fine-tuned version of [nvidia/mit-b0](https://huggingface.co/nvidia/mit-b0) on an unknown dataset.
|
| 16 |
It achieves the following results on the evaluation set:
|
| 17 |
+
- Train Loss: 0.0577
|
| 18 |
+
- Train Dice Coef: 0.8161
|
| 19 |
+
- Train Iou: 0.7082
|
| 20 |
+
- Validation Loss: 0.0458
|
| 21 |
- Validation Dice Coef: 0.8838
|
| 22 |
- Validation Iou: 0.7927
|
| 23 |
- Train Lr: 5e-06
|
| 24 |
+
- Epoch: 18
|
| 25 |
|
| 26 |
## Model description
|
| 27 |
|
|
|
|
| 65 |
| 0.0595 | 0.8160 | 0.7064 | 0.0475 | 0.8810 | 0.7883 | 5e-06 | 15 |
|
| 66 |
| 0.0589 | 0.8197 | 0.7115 | 0.0460 | 0.8835 | 0.7922 | 5e-06 | 16 |
|
| 67 |
| 0.0576 | 0.8214 | 0.7134 | 0.0459 | 0.8838 | 0.7927 | 5e-06 | 17 |
|
| 68 |
+
| 0.0577 | 0.8161 | 0.7082 | 0.0458 | 0.8838 | 0.7927 | 5e-06 | 18 |
|
| 69 |
|
| 70 |
|
| 71 |
### Framework versions
|
tf_model.h5
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 15133664
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:c92b72915a773a318c28b87cb13aef65fd84040a5742bc10ef92f41e60e0cd60
|
| 3 |
size 15133664
|