인공지능/딥러닝
딥러닝 전이학습
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