Spaces:
Sleeping
Sleeping
Upload folder using huggingface_hub
Browse files
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()
|