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

Method init_weights

models/NHits.py:74–78  ·  view source on GitHub ↗
(self)

Source from the content-addressed store, hash-verified

72 self.init_weights()
73
74 def init_weights(self):
75 self.causalconv1.conv.weight.data.normal_(0, 0.01)
76 self.causalconv2.conv.weight.data.normal_(0, 0.01)
77 if self.downsample is not None:
78 self.downsample.weight.data.normal_(0, 0.01)
79
80 def forward(self, x):
81 out = self.net(x)

Callers 1

__init__Method · 0.95

Calls

no outgoing calls

Tested by

no test coverage detected