1. pytorch 中冻结模型参数后参数仍会被调整

dododododoooo2024-11-09 11:09

问题

python 复制代码

        self.sgf_net.requires_grad_(False)

起初设置 requires_grad(False) 优化器也没有添加sgfnet的模型参数。但是在pylightning框架中，每次推理完模型的参数都会被改变，经过仔细排查发现问题

python 复制代码

        # self.sgf_net.requires_grad_(False)
        for param in self.sgf_net.parameters():
            param.requires_grad = False
        self.sgf_net.eval()
```
添加 model.eval() 设置为推理模型就可以了