'🖥️ Programming/Pytorch' 카테고리의 글 목록

[Pytorch] if name == 'main'

2023.09.05

파이썬 코드를 보면 `if __name__ == '__main__':` 이런 코드를 확인할 수 있다. 이는 파이썬 코드를 모듈로 사용할 때와 직접 실행할 때를 구분하기 위해 사용하는 코드 패턴으로. 해당 모듈이 임포트된 경우가 아니라 인터프리터에서 직접 실행된 경우에만 if문 코드를 실행하라는 명령어이다. 다음과 같은 코드 예시가 있다고 하자. # my_module.py def my_function(): print("모듈 내 함수 실행") if __name__ == '__main__': print("직접 실행됨") 여기서 다른 모듈 `another_module.py`에서 다음과 같이 `my_function()`이라는 함수를 호출하면 "모듈 내 함수 실행"이라는 실행 결과가 나온다. # another_mo..

🖥️ Programming/Pytorch

[Pytorch] 모델 학습 및 예측 코드 - torch.no_grad(), loss.backward(), optimizer.step(), optimizer_zero_grad()

2023.08.29

def training(epoch, model, trainloader, validloader): correct = 0 total = 0 running_loss = 0 model.train() for b in trainloader: x, y = b.text, b.label x, y = x.to(device), y.to(device) # 반드시 모델과 같은 device y_pred = model(x) loss = loss_fn(y_pred, y) optimizer.zero_grad() loss.backward() optimizer.step() with torch.no_grad(): # 추론 y_pred = torch.argmax(y_pred, dim=1) correct += (y_pred == y).sum(..

🖥️ Programming/Pytorch

[Pytorch] F.relu()와 nn.ReLU()의 차이점

2023.08.29

nn.ReLU() 파이토치의 신경망 모듈 중 하나인 활성화 함수 모듈 `nn.Sequential()` 모델에 추가할 수 있는 `nn.Module`을 만듬 모델의 `__init__` 메서드에서 레이어로 초기화되며, 그 레이어가 `forward()` 메서드에서 사용될 때 활성화 함수가 적용 F.relu() 파이토치의 함수 라이브러리인 `torch.nn.functional`의 함수 함수로서 주로 모델의 forward 연산 내에서 활성화 함수를 적용하는 데 사용 모델의 forward 메서드에서 함수로 직접 호출 https://discuss.pytorch.org/t/whats-the-difference-between-nn-relu-vs-f-relu/27599 https://asidefine.tistory.com/80

[Pytorch] if __name__ == '__main__'

[Pytorch] 모델 학습 및 예측 코드 - torch.no_grad(), loss.backward(), optimizer.step(), optimizer_zero_grad()

[Pytorch] F.relu()와 nn.ReLU()의 차이점

티스토리툴바

[Pytorch] if name == 'main'