| |
| 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' |
| ] |
|
|
