Fix unfreeze_and_add_param_group expects modules rather than module (#6822)

sadiqj · web-flow · commit 7f91c5ebbc60 · 2021-04-06T01:50:42.000+02:00
diff --git a/pytorch_lightning/callbacks/finetuning.py b/pytorch_lightning/callbacks/finetuning.py
@@ -77,7 +77,7 @@ def finetune_function(self, pl_module, current_epoch, optimizer, optimizer_idx):
                 # When `current_epoch` is 10, feature_extractor will start training.
                 if current_epoch == self._unfreeze_at_epoch:
                     self.unfreeze_and_add_param_group(
-                        module=pl_module.feature_extractor,
+                        modules=pl_module.feature_extractor,
                         optimizer=optimizer,
                         train_bn=True,
                     )

Original file line number	Diff line number	Diff line change
`@@ -77,7 +77,7 @@ def finetune_function(self, pl_module, current_epoch, optimizer, optimizer_idx):`
`77`	`77`	# When `current_epoch` is 10, feature_extractor will start training.
`78`	`78`	`if current_epoch == self._unfreeze_at_epoch:`
`79`	`79`	`self.unfreeze_and_add_param_group(`
`80`		`- module=pl_module.feature_extractor,`
	`80`	`+ modules=pl_module.feature_extractor,`
`81`	`81`	`optimizer=optimizer,`
`82`	`82`	`train_bn=True,`
`83`	`83`	`)`