AhamadShaik commited on
Commit
5817dbd
·
1 Parent(s): 8b5da51

Training in progress epoch 10

Browse files
Files changed (2) hide show
  1. README.md +8 -7
  2. 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.0315
18
- - Train Dice Coef: 0.7137
19
- - Train Iou: 0.5680
20
- - Validation Loss: 0.0307
21
- - Validation Dice Coef: 0.8333
22
- - Validation Iou: 0.7169
23
  - Train Lr: 1e-04
24
- - Epoch: 9
25
 
26
  ## Model description
27
 
@@ -57,6 +57,7 @@ The following hyperparameters were used during training:
57
  | 0.0340 | 0.7019 | 0.5531 | 0.0327 | 0.8297 | 0.7113 | 1e-04 | 7 |
58
  | 0.0339 | 0.7002 | 0.5530 | 0.0319 | 0.8344 | 0.7182 | 1e-04 | 8 |
59
  | 0.0315 | 0.7137 | 0.5680 | 0.0307 | 0.8333 | 0.7169 | 1e-04 | 9 |
 
60
 
61
 
62
  ### 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.0292
18
+ - Train Dice Coef: 0.7341
19
+ - Train Iou: 0.5908
20
+ - Validation Loss: 0.0296
21
+ - Validation Dice Coef: 0.8400
22
+ - Validation Iou: 0.7266
23
  - Train Lr: 1e-04
24
+ - Epoch: 10
25
 
26
  ## Model description
27
 
 
57
  | 0.0340 | 0.7019 | 0.5531 | 0.0327 | 0.8297 | 0.7113 | 1e-04 | 7 |
58
  | 0.0339 | 0.7002 | 0.5530 | 0.0319 | 0.8344 | 0.7182 | 1e-04 | 8 |
59
  | 0.0315 | 0.7137 | 0.5680 | 0.0307 | 0.8333 | 0.7169 | 1e-04 | 9 |
60
+ | 0.0292 | 0.7341 | 0.5908 | 0.0296 | 0.8400 | 0.7266 | 1e-04 | 10 |
61
 
62
 
63
  ### Framework versions
tf_model.h5 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:4892ea4dbd15ed57b7b237acdf5e8a584c7e109bbb706cce905ca107011ac0bf
3
  size 15133664
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4a132e101e39213ede8f381969789a377d9fcd64ebe1e6f74e2929a5ca3257a3
3
  size 15133664