Spaces:
Sleeping
Sleeping
Upload run_cloud_training.py with huggingface_hub
Browse files- run_cloud_training.py +2 -2
run_cloud_training.py
CHANGED
@@ -568,8 +568,8 @@ def train(config_path, dataset_name, output_dir, upload_to_hub=False, hub_repo_n
|
|
568 |
bf16 = torch.cuda.is_available() and hardware_config.get("bf16", False)
|
569 |
gradient_checkpointing = torch.cuda.is_available() and hardware_config.get("gradient_checkpointing", True)
|
570 |
dataloader_workers = training_config.get("dataloader_num_workers", 4)
|
571 |
-
eval_strategy = training_config.get("eval_strategy", "
|
572 |
-
load_best_model_at_end =
|
573 |
logger.info("Using full training parameters for GPU mode")
|
574 |
else:
|
575 |
# For Space CPU training mode, use minimal parameters
|
|
|
568 |
bf16 = torch.cuda.is_available() and hardware_config.get("bf16", False)
|
569 |
gradient_checkpointing = torch.cuda.is_available() and hardware_config.get("gradient_checkpointing", True)
|
570 |
dataloader_workers = training_config.get("dataloader_num_workers", 4)
|
571 |
+
eval_strategy = training_config.get("eval_strategy", "no") # Changed from "steps" to "no" since we don't have an eval_dataset
|
572 |
+
load_best_model_at_end = False # Must be False when eval_strategy is "no"
|
573 |
logger.info("Using full training parameters for GPU mode")
|
574 |
else:
|
575 |
# For Space CPU training mode, use minimal parameters
|