George-API commited on
Commit
7e2aaf9
·
verified ·
1 Parent(s): 0cd991c

Upload run_cloud_training.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. 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", "steps") # Updated from evaluation_strategy
572
- load_best_model_at_end = training_config.get("load_best_model_at_end", True)
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