Gracoy commited on
Commit
949d4df
1 Parent(s): 8903580

Upload model

Browse files
Files changed (2) hide show
  1. README.md +7 -6
  2. tf_model.h5 +1 -1
README.md CHANGED
@@ -13,9 +13,9 @@ probably proofread and complete it, then remove this comment. -->
13
 
14
  This model was trained from scratch on an unknown dataset.
15
  It achieves the following results on the evaluation set:
16
- - Train Loss: 1.0459
17
- - Validation Loss: 1.0613
18
- - Epoch: 1
19
 
20
  ## Model description
21
 
@@ -34,15 +34,16 @@ 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': 0.0003, 'decay': 0.0, 'beta_1': 0.9, 'beta_2': 0.99, 'epsilon': 1e-07, '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 | Validation Loss | Epoch |
43
  |:----------:|:---------------:|:-----:|
44
- | 1.0820 | 1.0663 | 0 |
45
- | 1.0459 | 1.0613 | 1 |
 
46
 
47
 
48
  ### Framework versions
 
13
 
14
  This model was trained from scratch on an unknown dataset.
15
  It achieves the following results on the evaluation set:
16
+ - Train Loss: 0.9775
17
+ - Validation Loss: 1.0329
18
+ - Epoch: 2
19
 
20
  ## Model description
21
 
 
34
  ### Training hyperparameters
35
 
36
  The following hyperparameters were used during training:
37
+ - optimizer: {'inner_optimizer': {'class_name': 'AdamWeightDecay', 'config': {'name': 'AdamWeightDecay', 'learning_rate': 0.0003, 'decay': 0.0, 'beta_1': 0.9, 'beta_2': 0.99, 'epsilon': 1e-07, 'amsgrad': True, '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 | Validation Loss | Epoch |
43
  |:----------:|:---------------:|:-----:|
44
+ | 1.0466 | 1.0321 | 0 |
45
+ | 1.0046 | 1.0296 | 1 |
46
+ | 0.9775 | 1.0329 | 2 |
47
 
48
 
49
  ### Framework versions
tf_model.h5 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:f0eface4406bf40a6986fe6415c9a53dcd26cbdd2ccf902ec6b3803586755673
3
  size 370269264
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ef9cf87ded1469cb70ea860f0e2a3028d5b2f1cf7f428758e0a6f436e92aa773
3
  size 370269264