Pytorch 实现冻结指定卷积层的参数
python代码
fori,parainenumerate(self._net.module.features.parameters()): ifi<16: para.requires_grad=False else: para.requires_grad=True #Solver. #self._solver=torch.optim.SGD( #self._net.parameters(),lr=self._options['base_lr'], #momentum=0.9,weight_decay=self._options['weight_decay']) self._solver=torch.optim.SGD( self._net.module.parameters(),lr=self._options['base_lr'], momentum=0.9,weight_decay=self._options['weight_decay'])
分析
通过for循环将需要冻结的layer的requires_grad属性设置为False
以上这篇Pytorch实现冻结指定卷积层的参数就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持毛票票。
声明：本文内容来源于网络，版权归原作者所有，内容由互联网用户自发贡献自行上传，本网站不拥有所有权，未作人工编辑处理，也不承担相关法律责任。如果您发现有涉嫌版权的内容，欢迎发送邮件至：czq8825#qq.com（发邮件时，请将#更换为@）进行举报，并提供相关证据，一经查实，本站将立刻删除涉嫌侵权内容。