Tweak comments (meta-pytorch#1565)

Summary: Also use integer for numerator in division. Pull Request resolved: meta-pytorch#1565 Reviewed By: cyrjano Differential Revision: D74943172 Pulled By: jjuncho fbshipit-source-id: 251b8ce1cc849de9aa39eb1c99910fc660064885

Author: j-t-1

captum/module/binary_concrete_stochastic_gates.py

@@ -58,7 +58,7 @@ def __init__(
  n_gates: int,
  mask: Optional[Tensor] = None,
  reg_weight: float = 1.0,
- temperature: float = 2.0 / 3,
+ temperature: float = 2 / 3,
  lower_bound: float = -0.1,
  upper_bound: float = 1.1,
  eps: float = 1e-8,
@@ -118,16 +118,17 @@ def __init__(
 
  assert (
  0 < temperature < 1
- ), f"the temperature should be bwteen 0 and 1, received {temperature}"
+ ), f"the temperature should be between 0 and 1, received {temperature}"
  self.temperature = temperature
 
  assert (
  lower_bound < 0
- ), f"the stretch lower bound should smaller than 0, received {lower_bound}"
+ ), f"the stretch lower bound should be smaller than 0, received {lower_bound}"
  self.lower_bound = lower_bound
+
  assert (
  upper_bound > 1
- ), f"the stretch upper bound should larger than 1, received {upper_bound}"
+ ), f"the stretch upper bound should be larger than 1, received {upper_bound}"
  self.upper_bound = upper_bound
 
  self.eps = eps