discrepancy update works, testing with models

desResLab · Nov 7, 2023 · 70a9e4e · 70a9e4e
1 parent 4864f30
commit 70a9e4e
Show file tree

Hide file tree

Showing 2 changed files with 403 additions and 2 deletions.
diff --git a/linfa/run_experiment.py b/linfa/run_experiment.py
@@ -303,6 +303,7 @@ def train(self, nf, optimizer, iteration, log, sampling=True, t=1):
                 print('Invalid type of surrogate model')
                 exit(-1)
             if(go_on):    
+                # Update Surrogate Model
                 if(self.surrogate_type == 'surrogate'):
                     xk0 = xk[:self.true_data_num, :].data.clone()
                     self.surrogate.update(xk0, max_iters=self.surr_upd_it)
@@ -312,8 +313,6 @@ def train(self, nf, optimizer, iteration, log, sampling=True, t=1):
                 else:
                     print('Invalid type of surrogate model')
                     exit(-1)    
-                # Update Surrogate Model                
-
 
         # Free energy bound
         loss = (- torch.sum(sum_log_abs_det_jacobians, 1) - t * self.model_logdensity(xk)).mean()