frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

Reland "Make adding buffers more like adding parameters (#104069)" (#106224)

Browse Source 
Pull Request resolved: https://github.com/pytorch/pytorch/pull/106224
Approved by: https://github.com/atalman, https://github.com/albanD
This commit is contained in:
Mikayla Gawarecki
2023-07-28 13:59:42 -07:00
committed by PyTorch MergeBot
parent 50e3f9cbbb
commit d8e5f2aa6d
40 changed files with 267 additions and 148 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
test/test_fx_experimental.py
View File

@ -1215,8 +1215,8 @@ class {test_classname}(torch.nn.Module):
self.seq = torch.nn.Sequential(torch.nn.BatchNorm1d(2, 2))
self.linear = torch.nn.Linear(2, 2)
self.attr = torch.randn(2)
self.register_buffer("attr2", torch.randn(2))
self.register_buffer("attr3", torch.ones(2, dtype=torch.int32))
self.attr2 = torch.nn.Buffer(torch.randn(2))
self.attr3 = torch.nn.Buffer(torch.ones(2, dtype=torch.int32))
def forward(self, x):
return self.linear(self.seq(self.W + self.attr + self.attr2 + self.attr3 + x))