从 Pytorch 中的 Resnet 模型中删除 RELU 激活

【问题标题】：Remove RELU activation from Resnet model in Pytorch从 Pytorch 中的 Resnet 模型中删除 RELU 激活
【发布时间】：2020-04-28 07:06:35
【问题描述】：

如何在 pytorch 中从 Resnet 模型中移除所有 RELU 激活层，并在可能的情况下将其替换为线性激活层？

【问题讨论】：

标签： python deep-learning pytorch

【解决方案1】：

model = models.resnet50()

names = []
for name, module in model.named_modules():
    if hasattr(module, 'relu'):
        module.relu = nn.Sigmoid() // or nn.Identity() accordingly

print(model)

这适用于替换激活或使其标识

【讨论】：

请将您的答案标记为用绿色勾号解决问题。