change parameter block

mieskolainen · Aug 1, 2024 · 1ff621b · 1ff621b
1 parent b86fe92
commit 1ff621b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/icenet/deep/optimize.py b/icenet/deep/optimize.py
@@ -198,7 +198,7 @@ def train(model, loader, optimizer, device, opt_param: dict, MI: dict=None, epoc
 
     sigma2 = None
     if 'noise_reg' in opt_param and opt_param['noise_reg'] > 0.0:
-        noise_reg = param['opt_param']['noise_reg']
+        noise_reg = opt_param['noise_reg']
         sigma2 = noise_reg * deeptools.sigmoid_schedule(t=opt_param['current_epoch'], N_max=opt_param['epochs'])
     # -------------------------------------------------------------------