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。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。