We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 8970c90 commit 2d76c45Copy full SHA for 2d76c45
1 file changed
swift/trainers/mixin.py
@@ -988,7 +988,8 @@ def create_optimizer_and_scheduler(self, num_training_steps: int):
988
989
def create_optimizer(self):
990
self.optimizer = self.optimizer_callback.create_optimizer()
991
- self.optimizer.param_groups = [pg for pg in self.optimizer.param_groups if len(pg['params']) > 0]
+ if self.optimizer is not None:
992
+ self.optimizer.param_groups = [pg for pg in self.optimizer.param_groups if len(pg['params']) > 0]
993
return self.optimizer
994
995
def create_scheduler(self, num_training_steps: int, optimizer):
0 commit comments