AhamadShaik commited on
Commit
2c33425
·
1 Parent(s): 78e9130

Training in progress epoch 36

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.0522
18
- - Train Dice Coef: 0.8408
19
- - Train Iou: 0.7385
20
- - Validation Loss: 0.0456
21
- - Validation Dice Coef: 0.8838
22
- - Validation Iou: 0.7927
23
  - Train Lr: 2.5e-07
24
- - Epoch: 35
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': 2.5e-07, 'decay': 0.0, 'beta_1': 0.9, 'beta_2': 0.999, 'epsilon': 1e-07, 'amsgrad': False}
44
  - training_precision: float32
45
 
46
  ### Training results
@@ -83,6 +83,7 @@ The following hyperparameters were used during training:
83
  | 0.0524 | 0.8414 | 0.7376 | 0.0453 | 0.8840 | 0.7930 | 2.5e-07 | 33 |
84
  | 0.0524 | 0.8406 | 0.7372 | 0.0446 | 0.8842 | 0.7935 | 2.5e-07 | 34 |
85
  | 0.0522 | 0.8408 | 0.7385 | 0.0456 | 0.8838 | 0.7927 | 2.5e-07 | 35 |
 
86
 
87
 
88
  ### 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.0521
18
+ - Train Dice Coef: 0.8484
19
+ - Train Iou: 0.7454
20
+ - Validation Loss: 0.0453
21
+ - Validation Dice Coef: 0.8839
22
+ - Validation Iou: 0.7929
23
  - Train Lr: 2.5e-07
24
+ - Epoch: 36
25
 
26
  ## Model description
27
 
 
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
 
83
  | 0.0524 | 0.8414 | 0.7376 | 0.0453 | 0.8840 | 0.7930 | 2.5e-07 | 33 |
84
  | 0.0524 | 0.8406 | 0.7372 | 0.0446 | 0.8842 | 0.7935 | 2.5e-07 | 34 |
85
  | 0.0522 | 0.8408 | 0.7385 | 0.0456 | 0.8838 | 0.7927 | 2.5e-07 | 35 |
86
+ | 0.0521 | 0.8484 | 0.7454 | 0.0453 | 0.8839 | 0.7929 | 2.5e-07 | 36 |
87
 
88
 
89
  ### Framework versions
tf_model.h5 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:837eb28a11ebe9f83abf35dfcf9ac6409db43c192f4b49afe74cd55289a5b47c
3
  size 15133664
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9b258edced2a1f15d7417d113dede2fcc3cf9008da3995601934f0e575d4064f
3
  size 15133664