AhamadShaik commited on
Commit
f771b70
·
1 Parent(s): 2539b1d

Training in progress epoch 7

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.0336
18
- - Train Dice Coef: 0.6770
19
- - Train Iou: 0.5307
20
- - Validation Loss: 0.0540
21
- - Validation Dice Coef: 0.6264
22
- - Validation Iou: 0.4881
23
  - Train Lr: 1e-04
24
- - Epoch: 6
25
 
26
  ## Model description
27
 
@@ -54,6 +54,7 @@ The following hyperparameters were used during training:
54
  | 0.0413 | 0.6306 | 0.4831 | 0.0771 | 0.5239 | 0.3646 | 1e-04 | 4 |
55
  | 0.0374 | 0.6569 | 0.5086 | 0.0719 | 0.5267 | 0.3854 | 1e-04 | 5 |
56
  | 0.0336 | 0.6770 | 0.5307 | 0.0540 | 0.6264 | 0.4881 | 1e-04 | 6 |
 
57
 
58
 
59
  ### 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.0302
18
+ - Train Dice Coef: 0.7029
19
+ - Train Iou: 0.5592
20
+ - Validation Loss: 0.0518
21
+ - Validation Dice Coef: 0.6516
22
+ - Validation Iou: 0.5234
23
  - Train Lr: 1e-04
24
+ - Epoch: 7
25
 
26
  ## Model description
27
 
 
54
  | 0.0413 | 0.6306 | 0.4831 | 0.0771 | 0.5239 | 0.3646 | 1e-04 | 4 |
55
  | 0.0374 | 0.6569 | 0.5086 | 0.0719 | 0.5267 | 0.3854 | 1e-04 | 5 |
56
  | 0.0336 | 0.6770 | 0.5307 | 0.0540 | 0.6264 | 0.4881 | 1e-04 | 6 |
57
+ | 0.0302 | 0.7029 | 0.5592 | 0.0518 | 0.6516 | 0.5234 | 1e-04 | 7 |
58
 
59
 
60
  ### Framework versions
tf_model.h5 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:925ccfe58c222e5c07f4df4f4e99e7d9b7609dbf8a5b2c65f9f6cf014be87cd5
3
  size 15133664
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bf183a0a9b4ea57d36875784c1e5bbdf88af7b5fb2cdcf8975827ab797a2b5e5
3
  size 15133664