PyTorch

훈련 시 반복문을 통해 dataloader에 접근할 때 pytorch에서는 다음과 같은 부분이 자주 쓴인다.model에 데이터를 입력하고 출력값을 pred_logit 변수에 담는다.실제 pred_logit의 shape은 bs, num_classes 이다.bs(batch

requires_grad = True이면 optimizer.step() 과정에서 업데이트 된다. requires_grad = False인 경우 업데이트 되지 않는다.nn.Moldule 클래스를 상속받은 모델의 레이어 및 파라미터 가중치를 확인하는 방법을 다룬다.