We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent c46df02 commit f602844Copy full SHA for f602844
captum/module/binary_concrete_stochastic_gates.py
@@ -134,8 +134,9 @@ def __init__(
134
self.eps = eps
135
136
# pre-calculate the fixed term used in active prob
137
- # pyre-fixme[4]: Attribute must be annotated.
138
- self.active_prob_offset = temperature * math.log(-lower_bound / upper_bound)
+ self.active_prob_offset: float = temperature * math.log(
+ -lower_bound / upper_bound
139
+ )
140
141
def _sample_gate_values(self, batch_size: int) -> Tensor:
142
"""
0 commit comments