AhamadShaik commited on
Commit
22ea2d6
·
1 Parent(s): 38dd4c1

Training in progress epoch 41

Browse files
Files changed (2) hide show
  1. README.md +9 -8
  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.0529
18
- - Train Dice Coef: 0.8410
19
- - Train Iou: 0.7388
20
- - Validation Loss: 0.0448
21
- - Validation Dice Coef: 0.8848
22
- - Validation Iou: 0.7944
23
  - Train Lr: 1.25e-08
24
- - Epoch: 40
25
 
26
  ## Model description
27
 
@@ -40,7 +40,7 @@ More information needed
40
  ### Training hyperparameters
41
 
42
  The following hyperparameters were used during training:
43
- - optimizer: {'name': 'Adam', 'learning_rate': 1.25e-08, 'decay': 0.0, 'beta_1': 0.9, 'beta_2': 0.999, 'epsilon': 1e-07, 'amsgrad': False}
44
  - training_precision: float32
45
 
46
  ### Training results
@@ -88,6 +88,7 @@ The following hyperparameters were used during training:
88
  | 0.0521 | 0.8370 | 0.7344 | 0.0451 | 0.8845 | 0.7939 | 1.25e-08 | 38 |
89
  | 0.0524 | 0.8484 | 0.7452 | 0.0456 | 0.8837 | 0.7927 | 1.25e-08 | 39 |
90
  | 0.0529 | 0.8410 | 0.7388 | 0.0448 | 0.8848 | 0.7944 | 1.25e-08 | 40 |
 
91
 
92
 
93
  ### 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.0519
18
+ - Train Dice Coef: 0.8402
19
+ - Train Iou: 0.7391
20
+ - Validation Loss: 0.0444
21
+ - Validation Dice Coef: 0.8852
22
+ - Validation Iou: 0.7951
23
  - Train Lr: 1.25e-08
24
+ - Epoch: 41
25
 
26
  ## Model description
27
 
 
40
  ### Training hyperparameters
41
 
42
  The following hyperparameters were used during training:
43
+ - optimizer: {'name': 'Adam', 'learning_rate': 6.25e-10, 'decay': 0.0, 'beta_1': 0.9, 'beta_2': 0.999, 'epsilon': 1e-07, 'amsgrad': False}
44
  - training_precision: float32
45
 
46
  ### Training results
 
88
  | 0.0521 | 0.8370 | 0.7344 | 0.0451 | 0.8845 | 0.7939 | 1.25e-08 | 38 |
89
  | 0.0524 | 0.8484 | 0.7452 | 0.0456 | 0.8837 | 0.7927 | 1.25e-08 | 39 |
90
  | 0.0529 | 0.8410 | 0.7388 | 0.0448 | 0.8848 | 0.7944 | 1.25e-08 | 40 |
91
+ | 0.0519 | 0.8402 | 0.7391 | 0.0444 | 0.8852 | 0.7951 | 1.25e-08 | 41 |
92
 
93
 
94
  ### Framework versions
tf_model.h5 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:139d8eea20fe128db32a38e16fe2affda4076552bfa675d84c220426bfeb75d7
3
  size 15133664
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:604a6ac5f025466476b6b2fff39294902dadd37369da4f0aa8eb7a0c69848abb
3
  size 15133664