Resultados da pesquisa a pedido "pytorch"

4 a resposta

Por que precisamos chamar explicitamente zero_grad ()?

Por que precisamos zerar explicitamente os gradientes no PyTorch? Por que os gradientes não podem ser zerados quandoloss.backward() é chamado? Qual cenário é servido mantendo os gradientes no gráfico e solicitando que o usuário ...

3 a resposta

Como inicializar pesos no PyTorch?

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

1 a resposta

Dispositivo inválido Ordinal, CUDA / TORCH

Estou recebendo esse erro ao executar o script no ubuntu 16.04. Por favor, tenha paciência comigo, sou novo no python, verifiquei as opções já disponíveis na internet, mas não consegui corrigi-lo. RuntimeError: cuda runtime error (10) : invalid ...

2 a resposta

Obtendo faixa matricial diagonal automaticamente em numpy ou pytorch

Preciso obter uma faixa diagonal da matriz (não tenho certeza sobre a terminologia aqui, a faixa diagonal da matriz parece descrevê-la melhor). Digamos, eu tenho uma matriz de tamanho KxN, onde K e N são tamanhos arbitrários e K> N. Diga, eu ...

2 a resposta

Implementando a desistência do zero

Este código tenta utilizar uma implementação personalizada de desistência: %reset -f import torch import torch.nn as nn # import torchvision # import torchvision.transforms as transforms import torch import torch.nn as nn import torch.utils.data ...

1 a resposta

Achatar a camada de construção do PyTorch por contêiner seqüencial

Estou tentando construir um cnn pelo contêiner seqüencial do PyTorch, meu problema é que não consigo descobrir como nivelar a camada. main = nn.Sequential() self._conv_block(main, 'conv_0', 3, 6, 5) main.add_module('max_pool_0_2_2', ...

1 a resposta

Como aplicar a taxa de aprendizado em camadas no Pytorch?

Eu sei que é possível congelar camadas únicas em uma rede, por exemplo, para treinar apenas as últimas camadas de um modelo pré-treinado. O que estou procurando é uma maneira de aplicar determinadas taxas de aprendizado a diferentes ...

1 a resposta

Por que a função de perda pode ser aplicada em diferentes tensores de tamanho

Por exemplo, eu tenho uma rede que toma o tensor [N, 7] (N é o número de amostras) como entrada e o tensor [N, 4] como saída, o "4" representa as probabilidades das diferentes classes. E os rótulos dos dados de treinamento são na forma de tensor ...

2 a resposta

Como você carrega imagens no Pytorch DataLoader?

O tutorial pytorch para carregamento e processamento de dados é bastante específico para um exemplo. Alguém poderia me ajudar com a aparência da função para um carregamento simples mais genérico de ...

1 a resposta

PyTorch - parâmetros não mudando

Em um esforço para aprender como funciona o pytorch, estou tentando fazer uma estimativa de probabilidade máxima de alguns dos parâmetros em uma distribuição normal multivariada. No entanto, parece não funcionar para nenhum dos ...