AhamadShaik commited on
Commit
cea9330
·
1 Parent(s): af7c708

Training in progress epoch 3

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.0582
18
- - Train Dice Coef: 0.5595
19
- - Train Iou: 0.4069
20
- - Validation Loss: 0.0509
21
- - Validation Dice Coef: 0.7745
22
- - Validation Iou: 0.6369
23
  - Train Lr: 1e-04
24
- - Epoch: 2
25
 
26
  ## Model description
27
 
@@ -50,6 +50,7 @@ The following hyperparameters were used during training:
50
  | 0.2695 | 0.2707 | 0.1687 | 0.1167 | 0.5713 | 0.4080 | 1e-04 | 0 |
51
  | 0.0805 | 0.4647 | 0.3202 | 0.0690 | 0.6569 | 0.4990 | 1e-04 | 1 |
52
  | 0.0582 | 0.5595 | 0.4069 | 0.0509 | 0.7745 | 0.6369 | 1e-04 | 2 |
 
53
 
54
 
55
  ### 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.0507
18
+ - Train Dice Coef: 0.5971
19
+ - Train Iou: 0.4430
20
+ - Validation Loss: 0.0459
21
+ - Validation Dice Coef: 0.7780
22
+ - Validation Iou: 0.6420
23
  - Train Lr: 1e-04
24
+ - Epoch: 3
25
 
26
  ## Model description
27
 
 
50
  | 0.2695 | 0.2707 | 0.1687 | 0.1167 | 0.5713 | 0.4080 | 1e-04 | 0 |
51
  | 0.0805 | 0.4647 | 0.3202 | 0.0690 | 0.6569 | 0.4990 | 1e-04 | 1 |
52
  | 0.0582 | 0.5595 | 0.4069 | 0.0509 | 0.7745 | 0.6369 | 1e-04 | 2 |
53
+ | 0.0507 | 0.5971 | 0.4430 | 0.0459 | 0.7780 | 0.6420 | 1e-04 | 3 |
54
 
55
 
56
  ### Framework versions
tf_model.h5 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:7a91db94ace5ddf37a5046f655e056780eb7e897b0808a0e9ddb0801d0b16f24
3
  size 15133664
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e91bf132428c2bde09c71febd571048bb52a621c62e397e5d83644663b9f44aa
3
  size 15133664