인공지능/딥러닝

딥러닝 전이학습

M.랄라 2023. 7. 9. 17:05
  • Model 파라미터 얼리기
    • model.named_parameters: 파라미터의 이름이 쭉 나옴
frozen = range(3, 60, 3)

for i, (name, param) in enumerate(model.named_parameters):

      if  i in frozen:

           param.required_grad = False