Zero-Shot Classification
Transformers
PyTorch
Safetensors
electra
text-classification
Inference Endpoints
saattrupdan commited on
Commit
5749a30
1 Parent(s): 038d16a

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -3
README.md CHANGED
@@ -67,10 +67,10 @@ You can use this model in your scripts as follows:
67
 
68
  The following hyperparameters were used during training:
69
  - learning_rate: 2e-05
70
- - train_batch_size: 2
71
- - eval_batch_size: 2
72
  - seed: 4242
73
- - gradient_accumulation_steps: 16
74
  - total_train_batch_size: 32
75
  - optimizer: Adam with betas=(0.9, 0.999) and epsilon=1e-08
76
  - lr_scheduler_type: linear
 
67
 
68
  The following hyperparameters were used during training:
69
  - learning_rate: 2e-05
70
+ - train_batch_size: 32
71
+ - eval_batch_size: 32
72
  - seed: 4242
73
+ - gradient_accumulation_steps: 1
74
  - total_train_batch_size: 32
75
  - optimizer: Adam with betas=(0.9, 0.999) and epsilon=1e-08
76
  - lr_scheduler_type: linear