Python mxnet.initializer.register用法及代码示例

用法: mxnet.initializer.register(klass)

参数：

klass：(class) - 的子类mx.init.Initializer需要注册为自定义初始化程序。

注册一个自定义初始化器。

可以通过扩展 mx.init.Initializer 并实现所需的函数(如 _init_weight 和 _init_bias )来创建自定义初始化程序。创建的初始化程序必须使用mx.init.register 注册，然后才能按名称调用。

示例：

>>> # Create and register a custom initializer that
... # initializes weights to 0.1 and biases to 1.
...
>>> @mx.init.register
... @alias('myinit')
... class CustomInit(mx.init.Initializer):
...   def __init__(self):
...     super(CustomInit, self).__init__()
...   def _init_weight(self, _, arr):
...     arr[:] = 0.1
...   def _init_bias(self, _, arr):
...     arr[:] = 1
...
>>> # Module is an instance of 'mxnet.module.Module'
...
>>> module.init_params("custominit")
>>> # module.init_params("myinit")
>>> # module.init_params(CustomInit())

相关用法

注：本文由纯净天空筛选整理自apache.org大神的英文原创作品 mxnet.initializer.register。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。