MCPcopy Create free account
hub / github.com/MegEngine/MegEngine / reset_parameters

Method reset_parameters

imperative/python/megengine/module/rnn.py:266–269  ·  view source on GitHub ↗
(self)

Source from the content-addressed store, hash-verified

264 self.reset_parameters()
265
266 def reset_parameters(self) -> None:
267 stdv = 1.0 / math.sqrt(self.hidden_size)
268 for weight in self.parameters():
269 init.uniform_(weight, -stdv, stdv)
270
271 @abstractmethod
272 def create_cell(self, layer):

Callers 2

_flatten_parametersMethod · 0.95
__init__Method · 0.45

Calls 2

parametersMethod · 0.80
sqrtMethod · 0.45

Tested by 1

__init__Method · 0.36