Skip to content

Commit 2d76c45

Browse files
committed
fix
1 parent 8970c90 commit 2d76c45

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

swift/trainers/mixin.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -988,7 +988,8 @@ def create_optimizer_and_scheduler(self, num_training_steps: int):
988988

989989
def create_optimizer(self):
990990
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]
991+
if self.optimizer is not None:
992+
self.optimizer.param_groups = [pg for pg in self.optimizer.param_groups if len(pg['params']) > 0]
992993
return self.optimizer
993994

994995
def create_scheduler(self, num_training_steps: int, optimizer):

0 commit comments

Comments
 (0)