Imene commited on
Commit
d6dec97
·
1 Parent(s): 5b93772

Training in progress epoch 0

Browse files
README.md CHANGED
@@ -14,13 +14,13 @@ probably proofread and complete it, then remove this comment. -->
14
 
15
  This model is a fine-tuned version of [google/vit-base-patch16-224](https://huggingface.co/google/vit-base-patch16-224) on an unknown dataset.
16
  It achieves the following results on the evaluation set:
17
- - Train Loss: 4.3094
18
- - Train Accuracy: 0.0226
19
- - Train Top-3-accuracy: 0.0668
20
- - Validation Loss: 4.4393
21
- - Validation Accuracy: 0.0226
22
- - Validation Top-3-accuracy: 0.0660
23
- - Epoch: 2
24
 
25
  ## Model description
26
 
@@ -39,16 +39,14 @@ More information needed
39
  ### Training hyperparameters
40
 
41
  The following hyperparameters were used during training:
42
- - optimizer: {'inner_optimizer': {'class_name': 'AdamWeightDecay', 'config': {'name': 'AdamWeightDecay', 'learning_rate': {'class_name': 'PolynomialDecay', 'config': {'initial_learning_rate': 0.003, 'decay_steps': 1400, '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}
43
  - training_precision: mixed_float16
44
 
45
  ### Training results
46
 
47
- | Train Loss | Train Accuracy | Train Top-3-accuracy | Validation Loss | Validation Accuracy | Validation Top-3-accuracy | Epoch |
48
  |:----------:|:--------------:|:--------------------:|:---------------:|:-------------------:|:-------------------------:|:-----:|
49
- | 4.6733 | 0.0106 | 0.0335 | 4.5751 | 0.0196 | 0.0464 | 0 |
50
- | 4.4185 | 0.0219 | 0.0647 | 4.3134 | 0.0242 | 0.0614 | 1 |
51
- | 4.3094 | 0.0226 | 0.0668 | 4.4393 | 0.0226 | 0.0660 | 2 |
52
 
53
 
54
  ### Framework versions
 
14
 
15
  This model is a fine-tuned version of [google/vit-base-patch16-224](https://huggingface.co/google/vit-base-patch16-224) on an unknown dataset.
16
  It achieves the following results on the evaluation set:
17
+ - Train Loss: 4.4859
18
+ - Train Accuracy: 0.0195
19
+ - Train Top-5-accuracy: 0.0579
20
+ - Validation Loss: 4.2995
21
+ - Validation Accuracy: 0.0368
22
+ - Validation Top-5-accuracy: 0.0865
23
+ - Epoch: 0
24
 
25
  ## Model description
26
 
 
39
  ### Training hyperparameters
40
 
41
  The following hyperparameters were used during training:
42
+ - optimizer: {'inner_optimizer': {'class_name': 'AdamWeightDecay', 'config': {'name': 'AdamWeightDecay', 'learning_rate': {'class_name': 'PolynomialDecay', 'config': {'initial_learning_rate': 0.0003, 'decay_steps': 1750, '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.001}}, 'dynamic': True, 'initial_scale': 32768.0, 'dynamic_growth_steps': 2000}
43
  - training_precision: mixed_float16
44
 
45
  ### Training results
46
 
47
+ | Train Loss | Train Accuracy | Train Top-5-accuracy | Validation Loss | Validation Accuracy | Validation Top-5-accuracy | Epoch |
48
  |:----------:|:--------------:|:--------------------:|:---------------:|:-------------------:|:-------------------------:|:-----:|
49
+ | 4.4859 | 0.0195 | 0.0579 | 4.2995 | 0.0368 | 0.0865 | 0 |
 
 
50
 
51
 
52
  ### Framework versions
logs/train/events.out.tfevents.1662926640.2114a1b34ca4.76.0.v2 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:71a7e4b2d5e9b2c23780910741859cf3f1996929bc050c6582d4da33274a715a
3
+ size 3266251
logs/validation/events.out.tfevents.1662927581.2114a1b34ca4.76.1.v2 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e469af1ad69f57b36204b5e95e16deec2fc70de55521ce7467b5c6a32cf946c6
3
+ size 530
tf_model.h5 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:ce7a1b83023e2a17f5c09bba8dc1f6bcd7456bad0e6f57a42cf54859a4a9a572
3
- size 343787296
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3a26438d7d5e1dc7cd92df35bf7bf8bd1a7aa73d040a6af8ddbc65c3841eda69
3
+ size 343770912