AhamadShaik commited on
Commit
8709491
1 Parent(s): 3aad26c

Training in progress epoch 21

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.0516
18
- - Train Dice Coef: 0.8632
19
- - Train Iou: 0.7617
20
- - Validation Loss: 0.0465
21
- - Validation Dice Coef: 0.8838
22
- - Validation Iou: 0.7934
23
  - Train Lr: 1e-04
24
- - Epoch: 20
25
 
26
  ## Model description
27
 
@@ -68,6 +68,7 @@ The following hyperparameters were used during training:
68
  | 0.0558 | 0.8544 | 0.7489 | 0.0476 | 0.8828 | 0.7917 | 1e-04 | 18 |
69
  | 0.0523 | 0.8617 | 0.7595 | 0.0454 | 0.8872 | 0.7983 | 1e-04 | 19 |
70
  | 0.0516 | 0.8632 | 0.7617 | 0.0465 | 0.8838 | 0.7934 | 1e-04 | 20 |
 
71
 
72
 
73
  ### 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.0515
18
+ - Train Dice Coef: 0.8636
19
+ - Train Iou: 0.7625
20
+ - Validation Loss: 0.0494
21
+ - Validation Dice Coef: 0.8816
22
+ - Validation Iou: 0.7894
23
  - Train Lr: 1e-04
24
+ - Epoch: 21
25
 
26
  ## Model description
27
 
 
68
  | 0.0558 | 0.8544 | 0.7489 | 0.0476 | 0.8828 | 0.7917 | 1e-04 | 18 |
69
  | 0.0523 | 0.8617 | 0.7595 | 0.0454 | 0.8872 | 0.7983 | 1e-04 | 19 |
70
  | 0.0516 | 0.8632 | 0.7617 | 0.0465 | 0.8838 | 0.7934 | 1e-04 | 20 |
71
+ | 0.0515 | 0.8636 | 0.7625 | 0.0494 | 0.8816 | 0.7894 | 1e-04 | 21 |
72
 
73
 
74
  ### Framework versions
tf_model.h5 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:8664619116f501421f823345233eb9f5bfb870d05bdc9870f133cf016c62f12e
3
  size 15133664
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0aaa53c6b6b834ddf7b814758a35f1e943918089e5da893d7a8f94108ed63ea5
3
  size 15133664