MCPcopy Create free account
hub / github.com/dcharatan/flowmap / init_weights_normal

Function init_weights_normal

flowmap/model/backbone/backbone_midas.py:17–22  ·  view source on GitHub ↗
(m)

Source from the content-addressed store, hash-verified

15
16def make_net(dims):
17 def init_weights_normal(m):
18 if type(m) == nn.Linear:
19 if hasattr(m, "weight"):
20 nn.init.kaiming_normal_(
21 m.weight, a=0.0, nonlinearity="relu", mode="fan_in"
22 )
23
24 layers = []
25 for i in range(len(dims) - 1):

Callers

nothing calls this directly

Calls

no outgoing calls

Tested by

no test coverage detected