kejian commited on
Commit
293becf
1 Parent(s): a8e5ec5

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.0001
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
@@ -136,4 +136,4 @@ The following hyperparameters were used during training:
136
  'weight_decay': 0.1}}
137
 
138
  # Wandb URL:
139
- https://wandb.ai/kejian/uncategorized/runs/304peotb
 
36
 
37
  The following hyperparameters were used during training:
38
  - learning_rate: 0.0001
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
 
136
  'weight_decay': 0.1}}
137
 
138
  # Wandb URL:
139
+ https://wandb.ai/kejian/uncategorized/runs/1gpnt88g