add super init to MilestonesFinetuning callback

Lightning-AI · May 6, 2021 · 0ca0b3c · 0ca0b3c
1 parent d52e0a8
commit 0ca0b3c
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