kejian commited on
Commit
fff7b69
1 Parent(s): 939e11d

update model card README.md

Browse files
Files changed (1) hide show
  1. README.md +4 -4
README.md CHANGED
@@ -36,10 +36,10 @@ More information needed
36
 
37
  The following hyperparameters were used during training:
38
  - learning_rate: 0.0005
39
- - train_batch_size: 32
40
- - eval_batch_size: 16
41
  - seed: 42
42
- - gradient_accumulation_steps: 4
43
  - total_train_batch_size: 128
44
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
45
  - lr_scheduler_type: linear
@@ -123,4 +123,4 @@ The following hyperparameters were used during training:
123
  'weight_decay': 0.1}}
124
 
125
  # Wandb URL:
126
- https://wandb.ai/kejian/uncategorized/runs/b4ejn68y
36
 
37
  The following hyperparameters were used during training:
38
  - learning_rate: 0.0005
39
+ - train_batch_size: 64
40
+ - eval_batch_size: 32
41
  - seed: 42
42
+ - gradient_accumulation_steps: 2
43
  - total_train_batch_size: 128
44
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
45
  - lr_scheduler_type: linear
123
  'weight_decay': 0.1}}
124
 
125
  # Wandb URL:
126
+ https://wandb.ai/kejian/uncategorized/runs/3se8ueja