MCPcopy Index your code
hub / github.com/pytorch/examples / reset_parameters

Method reset_parameters

distributed/FSDP2/model.py:71–73  ·  view source on GitHub ↗
(self)

Source from the content-addressed store, hash-verified

69 return self.resid_dropout(self.w2(self.gelu(self.w1(x))))
70
71 def reset_parameters(self):
72 self.w1.reset_parameters()
73 self.w2.reset_parameters()
74
75
76class TransformerBlock(nn.Module):

Callers

nothing calls this directly

Calls 1

reset_parametersMethod · 0.45

Tested by

no test coverage detected