swap log for log2 (shannon binary entropy)

main.py

@@ -18,7 +18,7 @@ model.eval()
 
 def compute_binary_entropy(p: torch.Tensor, eps: float = 1e-9) -> torch.Tensor:
     p = torch.clamp(p, min=eps, max=1.0 - eps)
-    return -(p * torch.log(p) + (1.0 - p) * torch.log(1.0 - p))
+    return -(p * torch.log2(p) + (1.0 - p) * torch.log2(1.0 - p))
 
 
 def compute_uncertainty(probs_samples: torch.Tensor, mean_probs: torch.Tensor) -> dict: