@@ -527,14 +527,14 @@ To summarize you can do:
|
|||||||
|
|
||||||
.. code-block:: python
|
.. code-block:: python
|
||||||
|
|
||||||
TrainingArguments(..., deespeed="/path/to/ds_config.json")
|
TrainingArguments(..., deepspeed="/path/to/ds_config.json")
|
||||||
|
|
||||||
or:
|
or:
|
||||||
|
|
||||||
.. code-block:: python
|
.. code-block:: python
|
||||||
|
|
||||||
ds_config_dict=dict(scheduler=scheduler_params, optimizer=optimizer_params)
|
ds_config_dict=dict(scheduler=scheduler_params, optimizer=optimizer_params)
|
||||||
TrainingArguments(..., deespeed=ds_config_dict)
|
TrainingArguments(..., deepspeed=ds_config_dict)
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user