AhamadShaik commited on
Commit
2a40c0b
·
1 Parent(s): 32da2a5

Training in progress epoch 45

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.0185
18
- - Train Dice Coef: 0.7775
19
- - Train Iou: 0.6493
20
- - Validation Loss: 0.0229
21
- - Validation Dice Coef: 0.8518
22
- - Validation Iou: 0.7456
23
  - Train Lr: 6.25e-10
24
- - Epoch: 44
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': 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
@@ -92,6 +92,7 @@ The following hyperparameters were used during training:
92
  | 0.0186 | 0.7788 | 0.6505 | 0.0231 | 0.8502 | 0.7434 | 6.25e-10 | 42 |
93
  | 0.0188 | 0.7773 | 0.6491 | 0.0232 | 0.8510 | 0.7443 | 6.25e-10 | 43 |
94
  | 0.0185 | 0.7775 | 0.6493 | 0.0229 | 0.8518 | 0.7456 | 6.25e-10 | 44 |
 
95
 
96
 
97
  ### 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.0187
18
+ - Train Dice Coef: 0.7765
19
+ - Train Iou: 0.6487
20
+ - Validation Loss: 0.0233
21
+ - Validation Dice Coef: 0.8491
22
+ - Validation Iou: 0.7416
23
  - Train Lr: 6.25e-10
24
+ - Epoch: 45
25
 
26
  ## Model description
27
 
 
40
  ### Training hyperparameters
41
 
42
  The following hyperparameters were used during training:
43
+ - optimizer: {'name': 'Adam', 'learning_rate': 1e-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
 
92
  | 0.0186 | 0.7788 | 0.6505 | 0.0231 | 0.8502 | 0.7434 | 6.25e-10 | 42 |
93
  | 0.0188 | 0.7773 | 0.6491 | 0.0232 | 0.8510 | 0.7443 | 6.25e-10 | 43 |
94
  | 0.0185 | 0.7775 | 0.6493 | 0.0229 | 0.8518 | 0.7456 | 6.25e-10 | 44 |
95
+ | 0.0187 | 0.7765 | 0.6487 | 0.0233 | 0.8491 | 0.7416 | 6.25e-10 | 45 |
96
 
97
 
98
  ### Framework versions
tf_model.h5 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:cb29eac1cac27570ec3b720db969e4cd6899991fbeaef259114e5356111fe43c
3
  size 15133664
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:47e9169322aadd904b271d2bcf8eb50418fe54bf3b4f7846e9682ad43abea92f
3
  size 15133664