MCPcopy Create free account
hub / github.com/K-Quant/HiDy / init_single_model

Function init_single_model

application/FFD/ffd.py:114–118  ·  view source on GitHub ↗
(model)

Source from the content-addressed store, hash-verified

112
113
114def init_single_model(model):
115 torch.nn.init.normal_(model.fc1.weight, mean = 0, std = 0.01)
116 torch.nn.init.normal_(model.fc2.weight, mean = 0, std = 0.01)
117 torch.nn.init.constant_(model.fc2.bias, val = 0)
118 torch.nn.init.constant_(model.fc2.bias, val = 0)
119
120
121

Callers

nothing calls this directly

Calls

no outgoing calls

Tested by

no test coverage detected