问题
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() 设置为推理模型就可以了