«pytorch» 태그된 질문

view()다음 코드 스 니펫에서 방법 에 대해 혼란 스럽습니다 . class Net(nn.Module): def __init__(self): super(Net, self).__init__() self.conv1 = nn.Conv2d(3, 6, 5) self.pool = nn.MaxPool2d(2,2) self.conv2 = nn.Conv2d(6, 16, 5) self.fc1 = nn.Linear(16*5*5, 120) self.fc2 = nn.Linear(120, 84) self.fc3 = nn.Linear(84, 10) def forward(self, x): x = self.pool(F.relu(self.conv1(x))) x = self.pool(F.relu(self.conv2(x))) …

205 python  memory  pytorch  torch  tensor 

PyTorch에서 훈련 된 모델을 저장하는 다른 방법을 찾고있었습니다. 지금까지 두 가지 대안을 찾았습니다. torch.save () 는 모델을 저장하고 torch.load () 는 모델을 저장합니다 . 훈련 된 모델을 저장하려면 model.state_dict () , 저장된 모델을로드하려면 model.load_state_dict () 접근법 1보다 접근법 2가 권장되는 이 토론을 보았습니다 . 내 질문은 왜 두 번째 접근 …

193 python  serialization  deep-learning  pytorch  tensor 

pytorch내 GPU를 사용 하고 있는지 알고 싶습니다 . nvidia-smi프로세스 중에 GPU에서 활동이 있는지 감지 할 수는 있지만 python스크립트로 작성된 것을 원합니다 . 그렇게 할 수있는 방법이 있습니까?

160 python  memory-management  gpu  nvidia  pytorch 

model.summary()방법이 있습니까? 다음과 같이 Keras에서 메서드가 수행하는 것처럼 PyTorch에서 모델의 요약을 인쇄 할 수 있습니까? Model Summary: ____________________________________________________________________________________________________ Layer (type) Output Shape Param # Connected to ==================================================================================================== input_1 (InputLayer) (None, 1, 15, 27) 0 ____________________________________________________________________________________________________ convolution2d_1 (Convolution2D) (None, 8, 15, 27) 872 input_1[0][0] ____________________________________________________________________________________________________ maxpooling2d_1 (MaxPooling2D) (None, 8, 7, 27) …

125 python  pytorch 

PyTorch의 네트워크에서 가중치와 편향을 초기화하는 방법 (예 : He 또는 Xavier 초기화 사용)?

116 python  neural-network  deep-learning  pytorch 

나는 PyTorch의 문서를 읽고 있고 그들이 쓰는 예제를 찾았습니다. gradients = torch.FloatTensor([0.1, 1.0, 0.0001]) y.backward(gradients) print(x.grad) 여기서 x는 y가 생성 된 초기 변수입니다 (3- 벡터). 문제는 그래디언트 텐서의 0.1, 1.0 및 0.0001 인수는 무엇입니까? 문서는 그것에 대해 명확하지 않습니다.

112 neural-network  gradient  pytorch  torch  gradient-descent 

이 메서드 zero_grad()는 훈련 중에 호출해야합니다. 그러나 문서 는별로 도움이되지 않습니다. | zero_grad(self) | Sets gradients of all model parameters to zero. 이 메서드를 호출해야하는 이유는 무엇입니까?

112 python  neural-network  deep-learning  pytorch  gradient-descent 

rnn에 대한 가변 길이 시퀀스 입력에 패킹을 사용하는 방법 을 복제하려고 했지만 먼저 시퀀스를 "포장"해야하는 이유를 이해해야합니다. 나는 우리가 그것들을 "패딩"해야하는 이유를 이해하지만 "패킹"(부터 pack_padded_sequence)이 필요한 이유는 무엇입니까? 높은 수준의 설명을 주시면 감사하겠습니다!

93 deep-learning  pytorch  recurrent-neural-network  tensor  zero-padding 

github (link) 에서 LSTM 언어 모델의이 예제를 살펴 보았습니다 . 일반적으로하는 일은 나에게 매우 분명합니다. 그러나 나는 여전히 contiguous()코드에서 여러 번 발생 하는 호출 이 무엇을하는지 이해하는 데 어려움을 겪고 있습니다. 예를 들어 74/75 행의 코드 입력 및 LSTM의 대상 시퀀스가 ​​생성됩니다. 데이터 (에 저장 됨 ids)는 2 차원이며 첫 …

90 neural-network  deep-learning  lstm  pytorch  contiguous 

numpy에서는 ndarray.reshape()배열을 재구성하는 데 사용 합니다. 나는 pytorch에서 사람들 torch.view(...)이 같은 목적을 위해 사용 하지만 동시에 torch.reshape(...)기존의 것도 있음을 알았습니다 . 그래서 나는 그들 사이의 차이점이 무엇이며 언제 둘 중 하나를 사용 해야하는지 궁금합니다.

83 pytorch 

신경망 모델의 매개 변수를 업데이트 / 변경하려고 시도한 후 업데이트 된 신경망의 전달 패스를 계산 그래프에 넣었습니다 (얼마나 많은 변경 / 업데이트가 있더라도). 나는이 아이디어를 시도했지만 그것을 할 때마다 pytorch는 업데이트 된 텐서 (모델 내부)를 리프로 설정하여 그라디언트를 받고 싶은 네트워크의 그라디언트 흐름을 죽입니다. 리프 노드는 내가 원하는 방식으로 계산 …

10 python  machine-learning  neural-network  pytorch 

다음과 같이 다음 CNN을 훈련하려고하지만 .cuda ()와 관련하여 동일한 오류가 계속 발생하며 수정 방법을 잘 모르겠습니다. 지금까지 내 코드 덩어리가 있습니다. import matplotlib.pyplot as plt import numpy as np import torch from torch import nn from torch import optim import torch.nn.functional as F import torchvision from torchvision import datasets, transforms, …

9 python  machine-learning  gpu  pytorch