Clean-up

utcsilab · Mar 19, 2023 · 68f7fb9 · 68f7fb9
1 parent 3f5f557
commit 68f7fb9
Show file tree

Hide file tree

Showing 4 changed files with 3 additions and 7 deletions.
diff --git a/aux_models.py b/aux_models.py
@@ -152,8 +152,7 @@ def forward(self, sample, num_unrolls):
 
                 # Travel a small amount
                 eps = torch.maximum(torch.amax(torch.abs(r), dim=(-1, -2),
-                                              keepdim=False) * 1e-3,
-                    self.safety_min)
+                                  keepdim=False) * 1e-3, self.safety_min)
                 if self.logging:
                     eps_log.append(copy.deepcopy(eps.cpu().detach().numpy()))
 

diff --git a/inference.sh b/inference.sh
diff --git a/test_l1Fourier_lifted.py b/test_l1Fourier_lifted.py
@@ -79,7 +79,7 @@
 complete_log = np.zeros((len(spacing_range), len(alpha_range),
                          len(lmbda_range), len(lr_range),
                          len(snr_range), gd_iter, kept_samples)) 
-result_dir = './results/l1_baseline_lifted%d/train-%s_test-%s' % (
+result_dir = './results/l1CS_lifted%d/train-%s_test-%s' % (
     lifting, args.train, args.test)
 os.makedirs(result_dir, exist_ok=True)
 

diff --git a/test_ldamp.py b/test_ldamp.py
@@ -48,7 +48,7 @@
 nmse_log     = np.zeros((len(spacing_range), len(pilot_alpha_range),
                          len(snr_range), num_channels))
 # Result directory
-result_dir = './results/ldamp_train-%s_test-%s' % (
+result_dir = './results/ldamp/train-%s_test-%s' % (
     args.train, args.test)
 os.makedirs(result_dir, exist_ok=True)