torch.nn.utils.remove_spectral_norm

torch.nn.utils.remove_spectral_norm(module, name='weight')[source][source]

从模块中移除光谱归一化的重新参数化

参数:

• 模块（Module）- 包含模块

• name（str，可选）- 权重参数的名称

返回类型:

模块 T

示例

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