core.models.equiformer_v2.weight_initialization

Contents

core.models.equiformer_v2.weight_initialization#

Copyright (c) Meta, Inc. and its affiliates.

This source code is licensed under the MIT license found in the LICENSE file in the root directory of this source tree.

Functions#

`eqv2_init_weights`(m, weight_init)
`eqv2_uniform_init_linear_weights`(m)

Module Contents#

core.models.equiformer_v2.weight_initialization.eqv2_init_weights(m, weight_init)#

core.models.equiformer_v2.weight_initialization.eqv2_uniform_init_linear_weights(m)#