MCPcopy Create free account
hub / github.com/BorealisAI/scaleformer / init_weights

Method init_weights

models/NHitsMS.py:88–92  ·  view source on GitHub ↗
(self)

Source from the content-addressed store, hash-verified

86 self.init_weights()
87
88 def init_weights(self):
89 self.causalconv1.conv.weight.data.normal_(0, 0.01)
90 self.causalconv2.conv.weight.data.normal_(0, 0.01)
91 if self.downsample is not None:
92 self.downsample.weight.data.normal_(0, 0.01)
93
94 def forward(self, x):
95 out = self.net(x)

Callers 1

__init__Method · 0.95

Calls

no outgoing calls

Tested by

no test coverage detected