Unverified Commit 3de12906 authored by Jake Tae's avatar Jake Tae Committed by GitHub
Browse files

fix: hfdeepspeed config argument (#15711)

`HfDeepSpeedConfig` accepts a dictionary or path to `.json` file containing DS configurations, not `TrainingArguments`.
parent 83f45cd6
......@@ -971,7 +971,7 @@ class Trainer:
# Rebuild the deepspeed config to reflect the updated training parameters
from transformers.deepspeed import HfDeepSpeedConfig
self.args.hf_deepspeed_config = HfDeepSpeedConfig(self.args)
self.args.hf_deepspeed_config = HfDeepSpeedConfig(self.args.deepspeed)
def _report_to_hp_search(
self, trial: Union["optuna.Trial", Dict[str, Any]], epoch: int, metrics: Dict[str, float]
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment