Call super().__init__() in MilestonesFinetuning example (#7398)

Lightning-AI · May 6, 2021 · d4d959b · d4d959b
1 parent a2a65ab
commit d4d959b
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/pl_examples/domain_templates/computer_vision_fine_tuning.py b/pl_examples/domain_templates/computer_vision_fine_tuning.py
@@ -69,6 +69,7 @@
 class MilestonesFinetuning(BaseFinetuning):
 
     def __init__(self, milestones: tuple = (5, 10), train_bn: bool = False):
+        super().__init__()
         self.milestones = milestones
         self.train_bn = train_bn