FatemehYp commited on
Commit
bd5330c
1 Parent(s): 0017525

Training in progress epoch 0

Browse files
Files changed (2) hide show
  1. README.md +4 -6
  2. tf_model.h5 +1 -1
README.md CHANGED
@@ -14,8 +14,8 @@ probably proofread and complete it, then remove this comment. -->
14
 
15
  This model is a fine-tuned version of [medicalai/ClinicalBERT](https://huggingface.co/medicalai/ClinicalBERT) on an unknown dataset.
16
  It achieves the following results on the evaluation set:
17
- - Train Loss: 1.3253
18
- - Epoch: 2
19
 
20
  ## Model description
21
 
@@ -34,16 +34,14 @@ More information needed
34
  ### Training hyperparameters
35
 
36
  The following hyperparameters were used during training:
37
- - optimizer: {'inner_optimizer': {'class_name': 'AdamWeightDecay', 'config': {'name': 'AdamWeightDecay', 'learning_rate': {'class_name': 'PolynomialDecay', 'config': {'initial_learning_rate': 2e-05, 'decay_steps': 444, 'end_learning_rate': 0.0, 'power': 1.0, 'cycle': False, 'name': None}}, 'decay': 0.0, 'beta_1': 0.9, 'beta_2': 0.999, 'epsilon': 1e-08, 'amsgrad': False, 'weight_decay_rate': 0.01}}, 'dynamic': True, 'initial_scale': 32768.0, 'dynamic_growth_steps': 2000}
38
  - training_precision: mixed_float16
39
 
40
  ### Training results
41
 
42
  | Train Loss | Epoch |
43
  |:----------:|:-----:|
44
- | 2.5750 | 0 |
45
- | 1.5897 | 1 |
46
- | 1.3253 | 2 |
47
 
48
 
49
  ### Framework versions
 
14
 
15
  This model is a fine-tuned version of [medicalai/ClinicalBERT](https://huggingface.co/medicalai/ClinicalBERT) on an unknown dataset.
16
  It achieves the following results on the evaluation set:
17
+ - Train Loss: 2.5710
18
+ - Epoch: 0
19
 
20
  ## Model description
21
 
 
34
  ### Training hyperparameters
35
 
36
  The following hyperparameters were used during training:
37
+ - optimizer: {'name': 'AdamWeightDecay', 'learning_rate': {'class_name': 'PolynomialDecay', 'config': {'initial_learning_rate': 2e-05, 'decay_steps': 444, 'end_learning_rate': 0.0, 'power': 1.0, 'cycle': False, 'name': None}}, 'decay': 0.0, 'beta_1': 0.9, 'beta_2': 0.999, 'epsilon': 1e-08, 'amsgrad': False, 'weight_decay_rate': 0.01}
38
  - training_precision: mixed_float16
39
 
40
  ### Training results
41
 
42
  | Train Loss | Epoch |
43
  |:----------:|:-----:|
44
+ | 2.5710 | 0 |
 
 
45
 
46
 
47
  ### Framework versions
tf_model.h5 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:9e707db8e90a5b66616dc7086a4c5f5b1dded74d570ccd8267011455616c417f
3
  size 539068392
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c8fba99ab404634cd9d4e843468299726820d383ee5403ae7a0a04ee720eefff
3
  size 539068392