Spaces:
Sleeping
Sleeping
Upload run_cloud_training.py with huggingface_hub
Browse files- run_cloud_training.py +1 -2
run_cloud_training.py
CHANGED
|
@@ -566,10 +566,9 @@ def train(config_path, dataset_name, output_dir):
|
|
| 566 |
"logging_first_step": training_config.get("logging_first_step", True),
|
| 567 |
"disable_tqdm": training_config.get("disable_tqdm", False),
|
| 568 |
"remove_unused_columns": False,
|
| 569 |
-
"shuffle": False, # Disable shuffling since we're using a pre-sorted dataset
|
| 570 |
"seed": 42
|
| 571 |
}
|
| 572 |
-
|
| 573 |
# Create TrainingArguments with validated parameters
|
| 574 |
training_args = TrainingArguments(**training_args_dict)
|
| 575 |
|
|
|
|
| 566 |
"logging_first_step": training_config.get("logging_first_step", True),
|
| 567 |
"disable_tqdm": training_config.get("disable_tqdm", False),
|
| 568 |
"remove_unused_columns": False,
|
|
|
|
| 569 |
"seed": 42
|
| 570 |
}
|
| 571 |
+
|
| 572 |
# Create TrainingArguments with validated parameters
|
| 573 |
training_args = TrainingArguments(**training_args_dict)
|
| 574 |
|