github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-04-29 03:00:14 -04:00
Code Issues Packages Projects Releases Wiki Activity

small cleanup

Browse Source
This commit is contained in:
Francis Lata
2024-09-27 13:28:53 -07:00
parent a879d19c82
commit 35105e769b
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/mlperf/initializers.py
View File

@@ -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):