728x90 pyTorch2 [딥러닝] Pytorch 모델 구조 및 파라미터 확인 파이토치(Pytorch)를 이용한 모델 학습 시 사용되는 모델의 구조, 파라미터(parameters)의 이름 및 개수를 확인하고 싶을 때 아래의 파이썬 코드를 통해 확인 가능하다. 방법 1 for parameter in model.parameters(): print(parameter) 방법 2 from torchsummary import summary summary(model, (1,)) # inputs: model, input_size=(1,) summary(model, (3,224,224)) # inputs: model, input_size=(3,224,224) 간단한 polynomial의 경우 입력의 차원은 (1,) 224x224 RGB 이미지의 경우 입력의 차원은 (3, 224, 224) 2022. 6. 25. [딥러닝] 우분투에서 CUDA 설정 & Pytorch 설치 경험상 Pytorch는 버전에 민감하다. 특정 버전의 Pytorch를 사용하기 위해 신중한 CUDA 버전 선택이 요구될 수 있다. 이번 글에서는 우분투 20.04에서 여러 CUDA 버전을 다운받아 그 중 한 경로를 설정하고 호환되는 Pytorch를 설치하는 방법에 대해 설명하겠다. 1. CUDA 설치 자신이 설치하고자 하는 Pytorch 버전에 호환되는 CUDA 버전을 확인한다. https://pytorch.org/get-started/previous-versions/ PyTorch An open source machine learning framework that accelerates the path from research prototyping to production deployment. pytor.. 2022. 4. 16. 이전 1 다음 728x90
