George-API commited on
Commit
0360950
·
verified ·
1 Parent(s): dc055e5

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. run_transformers_training.py +1 -1
run_transformers_training.py CHANGED
@@ -1040,7 +1040,7 @@ def main():
1040
  log_info("Cleared CUDA cache before training")
1041
 
1042
  # Display compact training info
1043
- total_steps = int(len(dataset) / (per_device_batch_size * NUM_GPUS * gradient_accumulation_steps) * training_args.num_train_epochs
1044
  log_info(f"Training plan: {len(dataset)} examples over {training_args.num_train_epochs} epochs ≈ {total_steps} steps")
1045
 
1046
  trainer.train()
 
1040
  log_info("Cleared CUDA cache before training")
1041
 
1042
  # Display compact training info
1043
+ total_steps = int((len(dataset) / (per_device_batch_size * NUM_GPUS * gradient_accumulation_steps)) * training_args.num_train_epochs)
1044
  log_info(f"Training plan: {len(dataset)} examples over {training_args.num_train_epochs} epochs ≈ {total_steps} steps")
1045
 
1046
  trainer.train()