評價此頁

torch.nn.utils.weight_norm.remove_weight_norm#

torch.nn.utils.weight_norm.remove_weight_norm(module, name='weight')[原始碼]#

從模組中移除權重歸一化重引數化。

引數
  • module (Module) – 包含的模組

  • name (str, optional) – 權重引數的名稱

返回型別

T_module

示例

>>> m = weight_norm(nn.Linear(20, 40))
>>> remove_weight_norm(m)