본문 바로가기

Data-science/deep learning

[pytorch] requires grad 확인.

728x90

named_parameters()루프를 돌면서. parameter의 requires_grad 변수를 확인해보면 된다.

True일 경우 해당 가중치는 학습한다. False일 경우 frozen, 얼려져있다. 학습하지 않는다.