Upload training_config.yml with huggingface_hub
Browse files- training_config.yml +2 -1
training_config.yml
CHANGED
|
@@ -29,6 +29,7 @@ temperature: 0.6
|
|
| 29 |
top_k: 300
|
| 30 |
dataset:
|
| 31 |
_component_: ds.EvenBatcher
|
|
|
|
| 32 |
dataset:
|
| 33 |
_component_: ds.RoundRobinDataset
|
| 34 |
datasets:
|
|
@@ -45,7 +46,7 @@ dataset:
|
|
| 45 |
train_on_input: false
|
| 46 |
seed: null
|
| 47 |
shuffle: true
|
| 48 |
-
batch_size:
|
| 49 |
optimizer:
|
| 50 |
_component_: torch.optim.AdamW
|
| 51 |
weight_decay: 0.01
|
|
|
|
| 29 |
top_k: 300
|
| 30 |
dataset:
|
| 31 |
_component_: ds.EvenBatcher
|
| 32 |
+
buffer_size: 36
|
| 33 |
dataset:
|
| 34 |
_component_: ds.RoundRobinDataset
|
| 35 |
datasets:
|
|
|
|
| 46 |
train_on_input: false
|
| 47 |
seed: null
|
| 48 |
shuffle: true
|
| 49 |
+
batch_size: 64
|
| 50 |
optimizer:
|
| 51 |
_component_: torch.optim.AdamW
|
| 52 |
weight_decay: 0.01
|