AhamadShaik commited on
Commit
21f2b08
·
1 Parent(s): 4099131

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.1105
18
- - Train Dice Coef: 0.7364
19
- - Train Iou: 0.5924
20
- - Validation Loss: 0.0737
21
- - Validation Dice Coef: 0.8147
22
- - Validation Iou: 0.6916
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.2282 | 0.5657 | 0.4102 | 0.1322 | 0.6524 | 0.4967 | 1e-04 | 0 |
51
  | 0.1354 | 0.6853 | 0.5329 | 0.0855 | 0.7853 | 0.6544 | 1e-04 | 1 |
52
  | 0.1105 | 0.7364 | 0.5924 | 0.0737 | 0.8147 | 0.6916 | 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.0985
18
+ - Train Dice Coef: 0.7610
19
+ - Train Iou: 0.6226
20
+ - Validation Loss: 0.0632
21
+ - Validation Dice Coef: 0.8518
22
+ - Validation Iou: 0.7440
23
  - Train Lr: 1e-04
24
+ - Epoch: 3
25
 
26
  ## Model description
27
 
 
50
  | 0.2282 | 0.5657 | 0.4102 | 0.1322 | 0.6524 | 0.4967 | 1e-04 | 0 |
51
  | 0.1354 | 0.6853 | 0.5329 | 0.0855 | 0.7853 | 0.6544 | 1e-04 | 1 |
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
 
55
 
56
  ### Framework versions
tf_model.h5 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:55e97ca7fd8750f298d966acb72beaf2162f8bb049fbefca17e5e2dfe8af70e0
3
  size 15133664
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f2ca8a9279b9fd9f0f8f9096187aac22125f38340d3f5947f291f13d6eaeede5
3
  size 15133664