La programmation pytorch

8

Comment fonctionne la méthode de «visualisation» dans PyTorch?

Je suis confus au sujet de la méthode view()dans l'extrait de code suivant. 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))) …

207 python memory pytorch torch tensor

5

Le meilleur moyen d'enregistrer un modèle entraîné dans PyTorch?

Je cherchais des moyens alternatifs pour enregistrer un modèle entraîné dans PyTorch. Jusqu'à présent, j'ai trouvé deux alternatives. torch.save () pour enregistrer un modèle et torch.load () pour charger un modèle. model.state_dict () pour enregistrer un modèle entraîné et model.load_state_dict () pour charger le modèle enregistré. Je suis tombé sur …

192 python serialization deep-learning pytorch tensor

10

Comment vérifier si pytorch utilise le GPU?

Je voudrais savoir si j'utilise pytorchmon GPU. Il est possible de détecter avec nvidia-smis'il y a une activité du GPU pendant le processus, mais je veux quelque chose d'écrit dans un pythonscript. Y a-t-il un moyen de le faire?

160 python memory-management gpu nvidia pytorch

11

Résumé du modèle dans Pytorch

Y a-t-il un moyen, je peux imprimer le résumé d'un modèle dans PyTorch comme la model.summary()méthode le fait dans Keras comme suit? 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

4

Pytorch, quels sont les arguments du gradient

Je lis la documentation de PyTorch et j'ai trouvé un exemple où ils écrivent gradients = torch.FloatTensor([0.1, 1.0, 0.0001]) y.backward(gradients) print(x.grad) où x était une variable initiale, à partir de laquelle y a été construit (un vecteur 3). La question est, quels sont les arguments 0,1, 1,0 et 0,0001 du …

112 neural-network gradient pytorch torch gradient-descent

2

Pourquoi avons-nous besoin d'appeler zero_grad () dans PyTorch?

La méthode zero_grad()doit être appelée pendant la formation. Mais la documentation n'est pas très utile | zero_grad(self) | Sets gradients of all model parameters to zero. Pourquoi devons-nous appeler cette méthode?

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

9

Comment initialiser les poids dans PyTorch?

Comment initialiser les poids et biais (par exemple, avec l'initialisation He ou Xavier) dans un réseau dans PyTorch?

106 python neural-network deep-learning pytorch

5

pourquoi «emballons-nous» les séquences en pytorch?

J'essayais de reproduire Comment utiliser l'empaquetage pour les entrées de séquence de longueur variable pour rnn, mais je suppose que je dois d'abord comprendre pourquoi nous devons "emballer" la séquence. Je comprends pourquoi nous devons les «tamponner» mais pourquoi est-il nécessaire de les «emballer» (à travers pack_padded_sequence)? Toute explication de …

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

6

PyTorch - contigu ()

Je passais par cet exemple de modèle de langage LSTM sur github (lien) . Ce qu'il fait en général est assez clair pour moi. Mais j'ai encore du mal à comprendre ce que fait l'appel contiguous(), ce qui se produit plusieurs fois dans le code. Par exemple, à la ligne …

90 neural-network deep-learning lstm pytorch contiguous

2

Comment peut-on avoir des paramètres dans un modèle pytorch qui ne soient pas des feuilles et qui soient dans le graphe de calcul?

J'essaie de mettre à jour / changer les paramètres d'un modèle de réseau neuronal et de faire ensuite passer le transfert vers l'avant du réseau neuronal mis à jour dans le graphique de calcul (peu importe le nombre de changements / mises à jour que nous faisons). J'ai essayé cette …

10 python machine-learning neural-network pytorch

1

RuntimeError: le type d'entrée (torch.FloatTensor) et le type de poids (torch.cuda.FloatTensor) doivent être identiques

J'essaie de former le CNN suivant comme suit, mais je reçois toujours la même erreur concernant .cuda () et je ne sais pas comment le corriger. Voici un morceau de mon code jusqu'à présent. import matplotlib.pyplot as plt import numpy as np import torch from torch import nn from torch …

9 python machine-learning gpu pytorch

Questions marquées «pytorch»