MCPcopy Create free account
hub / github.com/pytorch/examples / reset_parameters

Method reset_parameters

distributed/tensor_parallelism/llama2_model.py:141–142  ·  view source on GitHub ↗
(self)

Source from the content-addressed store, hash-verified

139 return output * self.weight
140
141 def reset_parameters(self):
142 torch.nn.init.ones_(self.weight) # type: ignore
143
144
145class Attention(nn.Module):

Callers 2

init_weightsMethod · 0.45
init_weightsMethod · 0.45

Calls

no outgoing calls

Tested by

no test coverage detected