update

open-mmlab · Mar 18, 2024 · 494c119 · 494c119
1 parent 94149da
commit 494c119
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/mmengine/_strategy/deepspeed.py b/mmengine/_strategy/deepspeed.py
@@ -553,8 +553,8 @@ def save_checkpoint(
 
         state_dict_kwargs = {}
         if digit_version(deepspeed.__version__) >= digit_version('0.13.2'):
-            state_dict_kwargs.update(
-                {'exclude_frozen_parameters': self.exclude_frozen_parameters})
+            state_dict_kwargs[
+                'exclude_frozen_parameters'] = self.exclude_frozen_parameters
 
         if save_optimizer:
             if hasattr(self, 'optim_wrapper'):