MCPcopy Create free account
hub / github.com/computational-imaging/bacon / init_weights_uniform

Function init_weights_uniform

modules.py:39–42  ·  view source on GitHub ↗
(m)

Source from the content-addressed store, hash-verified

37
38
39def init_weights_uniform(m):
40 if type(m) == nn.Linear:
41 if hasattr(m, 'weight'):
42 torch.nn.init.kaiming_uniform_(m.weight, nonlinearity='relu')
43
44
45def sine_init(m, w0=30):

Callers

nothing calls this directly

Calls

no outgoing calls

Tested by

no test coverage detected