評價此頁

torch.nn.utils.remove_spectral_norm#

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

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

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

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

返回型別

T_module

示例

>>> m = spectral_norm(nn.Linear(40, 10))
>>> remove_spectral_norm(m)