Perguntas com a marcação «pytorch»

Estou confuso sobre o método view()no seguinte trecho de código. 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 

Eu estava procurando maneiras alternativas de salvar um modelo treinado no PyTorch. Até agora, encontrei duas alternativas. torch.save () para salvar um modelo e torch.load () para carregar um modelo. model.state_dict () para salvar um modelo treinado e model.load_state_dict () para carregar o modelo salvo. Eu me deparei com essa …

192 python  serialization  deep-learning  pytorch  tensor 

Gostaria de saber se pytorchestá usando minha GPU. É possível detectar nvidia-smise há alguma atividade da GPU durante o processo, mas quero algo escrito em umpython script. Tem algum jeito de fazer isso?

160 python  memory-management  gpu  nvidia  pytorch 

Existe alguma maneira, eu posso imprimir o resumo de um modelo em PyTorch como o model.summary()método faz em Keras como segue? 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] ____________________________________________________________________________________________________ …

125 python  pytorch 

Estou lendo a documentação do PyTorch e encontrei um exemplo onde eles escrevem gradients = torch.FloatTensor([0.1, 1.0, 0.0001]) y.backward(gradients) print(x.grad) onde x era uma variável inicial, a partir da qual y foi construído (um 3-vetor). A questão é: quais são os argumentos de 0,1, 1,0 e 0,0001 do tensor de …

112 neural-network  gradient  pytorch  torch  gradient-descent 

Como inicializar os pesos e vieses (por exemplo, com inicialização de He ou Xavier) em uma rede em PyTorch?

106 python  neural-network  deep-learning  pytorch 

Eu estava tentando replicar como usar o empacotamento para entradas de sequência de comprimento variável para rnn, mas acho que primeiro preciso entender por que precisamos "empacotar" a sequência. Eu entendo por que precisamos "preenchê-los", mas por que "empacotar" (através pack_padded_sequence) é necessário? Qualquer explicação de alto nível seria apreciada!

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

Eu estava examinando este exemplo de um modelo de linguagem LSTM no github (link) . O que ele faz em geral é bastante claro para mim. Mas ainda estou lutando para entender o que a chamada contiguous()faz, o que ocorre várias vezes no código. Por exemplo, na linha 74/75 da …

94 neural-network  deep-learning  lstm  pytorch  contiguous 

Estou tentando atualizar / alterar os parâmetros de um modelo de rede neural e, em seguida, fazer com que a passagem direta da rede neural atualizada esteja no gráfico de computação (não importa quantas alterações / atualizações fazemos). Tentei essa idéia, mas sempre que faço isso, o pytorch define meus …

10 python  machine-learning  neural-network  pytorch 

Estou tentando treinar a seguinte CNN da seguinte maneira, mas continuo recebendo o mesmo erro em relação a .cuda () e não sei como corrigi-lo. Aqui está um pedaço do meu código até agora. import matplotlib.pyplot as plt import numpy as np import torch from torch import nn from torch …

9 python  machine-learning  gpu  pytorch 

Eu uso o Tensorflow, mas estou escrevendo documentação para usuários que normalmente variam entre as estruturas de aprendizado profundo . Ao trabalhar com conjuntos de dados que não se encaixam no sistema de arquivos local (TB +), colho dados de um repositório de dados remoto e gravo amostras localmente em …

8 python  pytorch 

Estou fazendo uma CNN com Pytorch para uma tarefa, mas ela não aprende e melhora a precisão. Eu fiz uma versão trabalhando com o conjunto de dados MNIST para que eu pudesse publicá-la aqui. Eu só estou procurando uma resposta sobre por que não está funcionando. A arquitetura é ótima, …

8 python  conv-neural-network  pytorch 

Existe um truque famoso na arquitetura u-net para usar mapas de peso personalizados para aumentar a precisão. Abaixo estão os detalhes: Agora, perguntando aqui e em vários outros lugares, conheço duas abordagens. Quero saber qual é a correta ou se existe outra abordagem correta que seja mais correta? 1) Primeiro …

8 python  pytorch  image-segmentation  semantic-segmentation