Merge branch 'master' of https://github.com/marsggbo/hyperbox

marsggbo · Jan 24, 2022 · 41b8dfc · 41b8dfc
2 parents d336cf2 + c9784ae
commit 41b8dfc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/hyperbox/networks/pytorch_modules.py b/hyperbox/networks/pytorch_modules.py
@@ -425,7 +425,7 @@ def forward(self, x):
         return topk_gumbel_softmax(x, self.tau, self.hard, self.topk, self.eps, self.dim)
 
 def topk_gumbel_softmax(logits: torch.Tensor, tau: float = 1, hard: bool = False,
-        topk: int = None, eps: float = 1e-10, dim: int = -1):
+        topk: int = 2, eps: float = 1e-10, dim: int = -1):
     # gumbels = (
     #     -torch.empty_like(
     #         logits, memory_format=torch.legacy_contiguous_format