5roop commited on
Commit
3ce07ee
1 Parent(s): b81f9a0

Update README with proper number of epochs

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -79,6 +79,6 @@ In fine-tuning, the following arguments were used:
79
  |-------------------------------|-------|
80
  | `per_device_train_batch_size` | 16 |
81
  | `gradient_accumulation_steps` | 4 |
82
- | `num_train_epochs` | 8 |
83
  | `learning_rate` | 3e-4 |
84
  | `warmup_steps` | 500 |
 
79
  |-------------------------------|-------|
80
  | `per_device_train_batch_size` | 16 |
81
  | `gradient_accumulation_steps` | 4 |
82
+ | `num_train_epochs` | 20 |
83
  | `learning_rate` | 3e-4 |
84
  | `warmup_steps` | 500 |