Domande taggate «pytorch»

Sono confuso sul metodo view()nel seguente frammento di codice. 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))) x …

206 python  memory  pytorch  torch  tensor 

Stavo cercando modi alternativi per salvare un modello addestrato in PyTorch. Finora ho trovato due alternative. torch.save () per salvare un modello e torch.load () per caricare un modello. model.state_dict () per salvare un modello addestrato e model.load_state_dict () per caricare il modello salvato. Mi sono imbattuto in questa discussione …

195 python  serialization  deep-learning  pytorch  tensor 

Vorrei sapere se pytorchsta usando la mia GPU. È possibile rilevare con nvidia-smise c'è qualche attività dalla GPU durante il processo, ma voglio qualcosa scritto in uno pythonscript. c'è un modo per farlo?

160 python  memory-management  gpu  nvidia  pytorch 

C'è un modo per stampare il riepilogo di un modello in PyTorch come il model.summary()metodo fa in Keras come 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] ____________________________________________________________________________________________________ maxpooling2d_1 …

125 python  pytorch 

Sto leggendo la documentazione di PyTorch e ho trovato un esempio in cui scrivono gradients = torch.FloatTensor([0.1, 1.0, 0.0001]) y.backward(gradients) print(x.grad) dove x era una variabile iniziale, da cui è stato costruito y (un 3-vettore). La domanda è: quali sono gli argomenti 0.1, 1.0 e 0.0001 del tensore dei gradienti? …

112 neural-network  gradient  pytorch  torch  gradient-descent 

Come inizializzare i pesi e i bias (ad esempio, con l'inizializzazione He o Xavier) in una rete in PyTorch?

106 python  neural-network  deep-learning  pytorch 

Stavo esaminando questo esempio di un modello di linguaggio LSTM su GitHub (collegamento) . Quello che fa in generale è abbastanza chiaro per me. Ma sto ancora lottando per capire cosa contiguous()fa la chiamata , che si verifica più volte nel codice. Ad esempio, nella riga 74/75 del codice di …

94 neural-network  deep-learning  lstm  pytorch  contiguous 

Sto cercando di aggiornare / modificare i parametri di un modello di rete neurale e quindi avere il passaggio in avanti della rete neurale aggiornata essere nel grafico di calcolo (indipendentemente da quanti cambiamenti / aggiornamenti facciamo). Ho provato questa idea ma ogni volta che lo faccio pytorch imposta i …

10 python  machine-learning  neural-network  pytorch 

Sto cercando di addestrare la seguente CNN come segue, ma continuo a ricevere lo stesso errore per quanto riguarda .cuda () e non sono sicuro di come risolverlo. Ecco un pezzo del mio codice finora. import matplotlib.pyplot as plt import numpy as np import torch from torch import nn from …

9 python  machine-learning  gpu  pytorch