зеркало из https://github.com/microsoft/DeepSpeed.git
Add default value to "checkpoint_folder" in "load_state_dict" of bf16_optimizer (#6446)
Add default value `checkpoint_folder=None` for compatibility. Co-authored-by: Olatunji Ruwase <olruwase@microsoft.com>
This commit is contained in:
Родитель
5d1a30c033
Коммит
ddd3571823
|
@ -482,7 +482,7 @@ class BF16_Optimizer(ZeROOptimizer):
|
|||
|
||||
def load_state_dict(self,
|
||||
state_dict_list,
|
||||
checkpoint_folder,
|
||||
checkpoint_folder=None,
|
||||
load_optimizer_states=True,
|
||||
load_from_fp32_weights=False,
|
||||
load_serial=None,
|
||||
|
|
Загрузка…
Ссылка в новой задаче