Pytorch实现冻结指定卷积层的参数-创新互联
python代码
成都网络公司-成都网站建设公司创新互联建站10余年经验成就非凡,专业从事成都做网站、网站制作,成都网页设计,成都网页制作,软文推广,一元广告等。10余年来已成功提供全面的成都网站建设方案,打造行业特色的成都网站建设案例,建站热线:028-86922220,我们期待您的来电!for i, para in enumerate(self._net.module.features.parameters()): if i < 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'])
另外有需要云服务器可以了解下创新互联scvps.cn,海内外云服务器15元起步,三天无理由+7*72小时售后在线,公司持有idc许可证,提供“云服务器、裸金属服务器、高防服务器、香港服务器、美国服务器、虚拟主机、免备案服务器”等云主机租用服务以及企业上云的综合解决方案,具有“安全稳定、简单易用、服务可用性高、性价比高”等特点与优势,专为企业上云打造定制,能够满足用户丰富、多元化的应用场景需求。
分享文章:Pytorch实现冻结指定卷积层的参数-创新互联
网站网址:http://ybzwz.com/article/djdiip.html