github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-09 06:58:11 -05:00
Code Issues Packages Projects Releases Wiki Activity

batchnorm d(var)/d(mean) = 0 (#4430)

Browse Source 
* d(var)/d(mean) = 0

* drop the number in test_schedule!
This commit is contained in:
David Hou
2024-05-04 21:25:45 -07:00
committed by GitHub
parent e2eab9c2b3
commit c0a048c044
3 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
test/test_schedule.py
View File

@@ -208,7 +208,7 @@ class TestSchedule(unittest.TestCase):
opt.zero_grad()
img_bn.backward()
# this is too high
check_schedule(opt.schedule_step(), 18)
check_schedule(opt.schedule_step(), 17)
def test_fold_conv_relu(self):
c1 = nn.Conv2d(3,16,3)