python程式碼

    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(),momentum=0.9,weight_decay=self._options['weight_decay'])

分析

通過for迴圈將需要凍結的layer的requires_grad屬性設定為False

以上這篇Pytorch 實現凍結指定卷積層的引數就是小編分享給大家的全部內容了，希望能給大家一個參考，也希望大家多多支援我們。