Update README to mention ICLR 2020 paper

HazyResearch · Aug 5, 2020 · 432c8d1 · 432c8d1
1 parent 64a5455
commit 432c8d1
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 4 deletions.
diff --git a/README.md b/README.md
@@ -1,4 +1,5 @@
-Code to accompany the paper <a href="https://arxiv.org/abs/1903.05895">Learning Fast Algorithms for Linear Transforms Using Butterfly Factorizations.
+Code to accompany the papers [Learning Fast Algorithms for Linear Transforms
+Using Butterfly Factorizations](https://arxiv.org/abs/1903.05895) and [Kaleidoscope: An Efficient, Learnable Representation For All Structured Linear Maps](https://openreview.net/forum?id=BkgrBgSYDS).
 
 ## Requirements
 python>=3.6  

diff --git a/tests/test_special.py b/tests/test_special.py
@@ -78,7 +78,6 @@ def test_circulant(self):
                 out = b(input)
                 self.assertTrue(torch.allclose(out, out_torch, self.rtol, self.atol))
 
-
     def test_hadamard(self):
         batch_size = 10
         n = 16

diff --git a/torch_butterfly/permutation.py b/torch_butterfly/permutation.py
@@ -299,5 +299,3 @@ def perm_to_butterfly_module(v, check=False):
         result = module(torch.eye(n))
         assert (torch.norm(result - torch.tensor(perm_vec_to_mat(invert(v))).float()).item() < 1e-6)
     return module
-
-
Original file line number	Diff line number	Diff line change
Expand Up		@@ -299,5 +299,3 @@ def perm_to_butterfly_module(v, check=False):
		result = module(torch.eye(n))
		assert (torch.norm(result - torch.tensor(perm_vec_to_mat(invert(v))).float()).item() < 1e-6)
		return module