p

nmcardoso · Aug 20, 2023 · efeb061 · efeb061
1 parent bf360c1
commit efeb061
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mergernet/estimators/parametric.py b/mergernet/estimators/parametric.py
@@ -98,7 +98,7 @@ def train(
       )
 
       # t1 train
-      if self.hp.get('t1_epochs') > 0:
+      if self.hp.get('t1_epochs', default=0) > 0:
         model = self.build(freeze_conv=True)
 
         opt = self.get_optimizer(self.hp.get('t1_opt'), lr=self.hp.get('t1_lr'))