20 lines
1023 B
Python
20 lines
1023 B
Python
# Copyright (c) OpenMMLab. All rights reserved.
|
|
from .flops_counter import get_model_complexity_info
|
|
from .fuse_conv_bn import fuse_conv_bn
|
|
from .sync_bn import revert_sync_batchnorm
|
|
from .weight_init import (INITIALIZERS, Caffe2XavierInit, ConstantInit,
|
|
KaimingInit, NormalInit, PretrainedInit,
|
|
TruncNormalInit, UniformInit, XavierInit,
|
|
bias_init_with_prob, caffe2_xavier_init,
|
|
constant_init, initialize, kaiming_init, normal_init,
|
|
trunc_normal_init, uniform_init, xavier_init)
|
|
|
|
__all__ = [
|
|
'get_model_complexity_info', 'bias_init_with_prob', 'caffe2_xavier_init',
|
|
'constant_init', 'kaiming_init', 'normal_init', 'trunc_normal_init',
|
|
'uniform_init', 'xavier_init', 'fuse_conv_bn', 'initialize',
|
|
'INITIALIZERS', 'ConstantInit', 'XavierInit', 'NormalInit',
|
|
'TruncNormalInit', 'UniformInit', 'KaimingInit', 'PretrainedInit',
|
|
'Caffe2XavierInit', 'revert_sync_batchnorm'
|
|
]
|