1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
|
from . import parametrizations, rnn, stateless
from .clip_grad import (
_clip_grads_with_norm_ as clip_grads_with_norm_,
_get_total_norm as get_total_norm,
clip_grad_norm,
clip_grad_norm_,
clip_grad_value_,
)
from .convert_parameters import parameters_to_vector, vector_to_parameters
from .fusion import (
fuse_conv_bn_eval,
fuse_conv_bn_weights,
fuse_linear_bn_eval,
fuse_linear_bn_weights,
)
from .init import skip_init
from .memory_format import (
convert_conv2d_weight_memory_format,
convert_conv3d_weight_memory_format,
)
from .spectral_norm import remove_spectral_norm, spectral_norm
from .weight_norm import remove_weight_norm, weight_norm
__all__ = [
"clip_grad_norm",
"clip_grad_norm_",
"clip_grads_with_norm_",
"clip_grad_value_",
"convert_conv2d_weight_memory_format",
"convert_conv3d_weight_memory_format",
"fuse_conv_bn_eval",
"fuse_conv_bn_weights",
"fuse_linear_bn_eval",
"fuse_linear_bn_weights",
"get_total_norm",
"parameters_to_vector",
"parametrizations",
"remove_spectral_norm",
"remove_weight_norm",
"rnn",
"skip_init",
"spectral_norm",
"stateless",
"vector_to_parameters",
"weight_norm",
]
|
