merge

yala · Nov 25, 2020 · 2ffc9af · 2ffc9af
2 parents 5466e81 + aab661f
commit 2ffc9af
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/onconet/models/mirai_full.py b/onconet/models/mirai_full.py
@@ -16,7 +16,7 @@ def __init__(self, args):
         else:
             self.image_encoder = get_model_by_name('custom_resnet', False, args)
 
-        if self.args.freeze_image_encoder:
+        if hasattr(self.args, "freeze_image_encoder") and self.args.freeze_image_encoder:
             for param in self.image_encoder.parameters():
                 param.requires_grad = False