MCPcopy Create free account
hub / github.com/JMoonr/LATR / init_weights

Method init_weights

models/transformer_bricks.py:397–400  ·  view source on GitHub ↗
(self)

Source from the content-addressed store, hash-verified

395 self.enlarge_length = enlarge_length
396
397 def init_weights(self):
398 super().init_weights()
399 for l in self.gflat_pred_layer:
400 xavier_init(l, gain=0.01)
401
402 def pred2M(self, pitch_z):
403 pitch_z = pitch_z / self.M_decay_ratio

Callers

nothing calls this directly

Calls 1

init_weightsMethod · 0.45

Tested by

no test coverage detected