mirror of
https://github.com/huggingface/transformers.git
synced 2025-08-02 19:21:31 +06:00
parent
98d5b72727
commit
8ea994d3c5
@ -778,6 +778,7 @@ class TrainerIntegrationDeepSpeed(TrainerIntegrationDeepSpeedWithCustomConfig, T
|
||||
"load_best_model_at_end": True,
|
||||
"max_steps": 1,
|
||||
"deepspeed": ds_config_dict,
|
||||
"report_to": "none",
|
||||
}
|
||||
|
||||
training_args = TrainingArguments(output_dir, **args_dict)
|
||||
|
Loading…
Reference in New Issue
Block a user