코생코사

tensor shape 변환요소의 수를 맞추는 것이 중요!! 안 맞추면 에러-1을 이용하여 알아서 계산하게 할 수 있다1. view()# 3X3X3data = torch.rand(3, 3, 3) # 총 요소 수 27개# 3X9로 바꾸기data.view(3, 9)# 행은 3으로, 열은 알아서 계산data.view(3, -1)# 1차원으로data.view(-1)2. reshape()# 3X3X3data = torch.rand(3, 3, 3) # 총 요소 수 27개# 3X9로 바꾸기data.reshape(3, 9)# 행은 3으로, 열은 알아서 계산data.reshape(3, -1)# 1차원으로data.reshape(-1) tensor의 형태 확인1. shape 확인data.shape2. 차원 확인data.di..

torch 모델을 학습시키고 저장하거나 불러오는 방법입니다 필요 라이브러리 임포트import torch  torch 모델 저장torch.save() 두 가지 방법이 있습니다.1. 모델 전체 저장하기(용량이 크다)2. 모델 가중치만 저장하기(용량이 작다) 1. 모델 전체 저장하기# 모델, 저장할 이름.pthtorch.save(model, 'model_all.pth') 2. 모델 가중치만 저장하기(권장)# 모델 가중치, 저장할 이름.pthtorch.save(model.state_dict(), 'model_weight.pth')   torch 모델 불러오기torch.load()두 방법 모두 모델 구조를 정의하는 클래스가 필요합니다!  모델 전체를 저장한 pth 파일 불러오기# 모델 저장 이름model = to..

쿠다 설치https://developer.nvidia.com/cuda-downloads CUDA Toolkit 12.1 DownloadsGet the latest feature updates to NVIDIA's proprietary compute stack.developer.nvidia.com버전 선택을 잘 해야 합니다지난 버전을 선택하고 싶다면 빨간 네모 박스 클릭 쿠다 버전에 맞게 터미널에서 파이토치 설치https://pytorch.org/ PyTorch pytorch.org자신의 컴퓨터 상황에 맞게 체크를 하고 command를 복사합니다.  cmd창에 붙여넣기pip3 install torch torchvision torchaudio --index-url https://download.pytorch..