Edit to allow unit-identification only

neuro-galaxy · Nov 29, 2024 · 2bad393 · 2bad393
1 parent 92a9c86
commit 2bad393
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/poyo_plus/finetune.py b/examples/poyo_plus/finetune.py
@@ -36,7 +36,7 @@ class GradualUnfreezing(L.Callback):
     frozen_params: Optional[List[nn.Parameter]] = None
 
     def __init__(self, unfreeze_at_epoch: int):
-        self.enabled = unfreeze_at_epoch > 0
+        self.enabled = unfreeze_at_epoch != 0
         self.unfreeze_at_epoch = unfreeze_at_epoch
         self.cli_log = logging.getLogger(__name__)