AhamadShaik commited on
Commit
717a6f0
1 Parent(s): e60ea41

Training in progress epoch 5

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.0933
18
- - Train Dice Coef: 0.7745
19
- - Train Iou: 0.6399
20
- - Validation Loss: 0.0627
21
- - Validation Dice Coef: 0.8455
22
- - Validation Iou: 0.7351
23
  - Train Lr: 1e-04
24
- - Epoch: 4
25
 
26
  ## Model description
27
 
@@ -52,6 +52,7 @@ The following hyperparameters were used during training:
52
  | 0.1105 | 0.7364 | 0.5924 | 0.0737 | 0.8147 | 0.6916 | 1e-04 | 2 |
53
  | 0.0985 | 0.7610 | 0.6226 | 0.0632 | 0.8518 | 0.7440 | 1e-04 | 3 |
54
  | 0.0933 | 0.7745 | 0.6399 | 0.0627 | 0.8455 | 0.7351 | 1e-04 | 4 |
 
55
 
56
 
57
  ### 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.0886
18
+ - Train Dice Coef: 0.7856
19
+ - Train Iou: 0.6535
20
+ - Validation Loss: 0.0584
21
+ - Validation Dice Coef: 0.8603
22
+ - Validation Iou: 0.7566
23
  - Train Lr: 1e-04
24
+ - Epoch: 5
25
 
26
  ## Model description
27
 
 
52
  | 0.1105 | 0.7364 | 0.5924 | 0.0737 | 0.8147 | 0.6916 | 1e-04 | 2 |
53
  | 0.0985 | 0.7610 | 0.6226 | 0.0632 | 0.8518 | 0.7440 | 1e-04 | 3 |
54
  | 0.0933 | 0.7745 | 0.6399 | 0.0627 | 0.8455 | 0.7351 | 1e-04 | 4 |
55
+ | 0.0886 | 0.7856 | 0.6535 | 0.0584 | 0.8603 | 0.7566 | 1e-04 | 5 |
56
 
57
 
58
  ### Framework versions
tf_model.h5 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:fa9ee5be962ce4e2ad0d2626939022babb94c042bc80ba9d8fdf28c9b34b248e
3
  size 15133664
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:075a502ffb1230a869e651c81a65bc62ce2a77671a87977a1cabde87c8ccb9d4
3
  size 15133664