adding freeze parameters

a-rouxel · Mar 9, 2024 · 9faab34 · 9faab34
1 parent f6cd82d
commit 9faab34
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/optimization_modules_with_resnet.py b/optimization_modules_with_resnet.py
@@ -41,6 +41,9 @@ def __init__(self, model_name,log_dir="tb_logs",reconstruction_checkpoint=None):
 
         self.writer = SummaryWriter(log_dir)
 
+        # for param in self.reconstruction_model.parameters():
+        #     param.requires_grad = False
+
     def on_validation_start(self,stage=None):
         print("---VALIDATION START---")
         self.config = "simca/configs/cassi_system_optim_optics_full_triplet_dd_cassi.yml"