diff --git a/examples/mlperf/initializers.py b/examples/mlperf/initializers.py index f8a2082853..a5d378550a 100644 --- a/examples/mlperf/initializers.py +++ b/examples/mlperf/initializers.py @@ -81,7 +81,7 @@ class Conv2dNormal(nn.Conv2d): if bias: if prior_prob: prior_prob = Tensor(prior_prob, device=self.bias.device, dtype=self.bias.dtype).expand(*self.bias.shape) - self.bias = (-(((1 - prior_prob) / prior_prob).log())) + self.bias = -(((1 - prior_prob) / prior_prob).log()) else: self.bias = Tensor.zeros_like(self.bias) class Conv2dKaimingUniform(nn.Conv2d):