[PyTorch] requires_grad 확인

requires_grad

• requires_grad = True이면 optimizer.step() 과정에서 업데이트 된다.
• requires_grad = False인 경우 업데이트 되지 않는다.

nn.Moldule 클래스를 상속받은 모델의 레이어 및 파라미터 가중치를 확인하는 방법을 다룬다.

# 파라미터 값을 출력
for param_value in model.parameters():
	print(param_value)

# requires_grad 출력
for param in model.parameters():
	print(param.requires_grad)

# name과 requires_grad 출력
for name param in model.named_parameters():
	print(name, param.requires_grad)