mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
ad4472833c
Adding modules imported here and the following functions to the `__all__`: * [clip_grad_norm_](https://pytorch.org/docs/stable/generated/torch.nn.utils.clip_grad_norm_.html) * [clip_grad_value_](https://pytorch.org/docs/stable/generated/torch.nn.utils.clip_grad_value_.html) * [remove_weight_norm](https://pytorch.org/docs/stable/generated/torch.nn.utils.remove_weight_norm.html) * [parameters_to_vector](https://pytorch.org/docs/stable/generated/torch.nn.utils.parameters_to_vector.html) * [vector_to_parameters](https://pytorch.org/docs/stable/generated/torch.nn.utils.vector_to_parameters.html) * [remove_spectral_norm](https://pytorch.org/docs/stable/generated/torch.nn.utils.remove_spectral_norm.html) * [skip_init](https://pytorch.org/docs/stable/generated/torch.nn.utils.skip_init.html) Pull Request resolved: https://github.com/pytorch/pytorch/pull/111026 Approved by: https://github.com/mikaylagawarecki
32 lines
993 B
Python
32 lines
993 B
Python
from . import rnn
|
|
from .clip_grad import clip_grad_norm, clip_grad_norm_, clip_grad_value_
|
|
from .weight_norm import weight_norm, remove_weight_norm
|
|
from .convert_parameters import parameters_to_vector, vector_to_parameters
|
|
from .spectral_norm import spectral_norm, remove_spectral_norm
|
|
from .fusion import fuse_conv_bn_eval, fuse_conv_bn_weights, fuse_linear_bn_eval, fuse_linear_bn_weights
|
|
from .memory_format import convert_conv2d_weight_memory_format
|
|
from . import parametrizations
|
|
from .init import skip_init
|
|
from . import stateless
|
|
|
|
__all__ = [
|
|
"clip_grad_norm",
|
|
"clip_grad_norm_",
|
|
"clip_grad_value_",
|
|
"convert_conv2d_weight_memory_format",
|
|
"fuse_conv_bn_eval",
|
|
"fuse_conv_bn_weights",
|
|
"fuse_linear_bn_eval",
|
|
"fuse_linear_bn_weights",
|
|
"parameters_to_vector",
|
|
"parametrizations",
|
|
"remove_spectral_norm",
|
|
"remove_weight_norm",
|
|
"rnn",
|
|
"skip_init",
|
|
"spectral_norm",
|
|
"stateless",
|
|
"vector_to_parameters",
|
|
"weight_norm",
|
|
]
|