Вопросы с тегом «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))) x = …

208 python  memory  pytorch  torch  tensor 

Я искал альтернативные способы сохранить обученную модель в PyTorch. Пока что я нашел две альтернативы. torch.save () для сохранения модели и torch.load () для загрузки модели. model.state_dict () для сохранения обученной модели и model.load_state_dict () для загрузки сохраненной модели. Я сталкивался с этим обсуждением, где подход 2 рекомендуется по подходу …

195 python  serialization  deep-learning  pytorch  tensor 

Я хотел бы знать, pytorchиспользует ли мой графический процессор. Можно определить, nvidia-smiесть ли какие-либо действия с графическим процессором во время процесса, но я хочу что-то записать в pythonсценарии. Есть ли способ сделать это?

160 python  memory-management  gpu  nvidia  pytorch 

Есть ли способ распечатать сводку модели в PyTorch, как model.summary()метод в Keras, как показано ниже? 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) …

127 python  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 

Я пытался воспроизвести, как использовать упаковку для входных данных последовательности переменной длины для rnn, но я думаю, что сначала мне нужно понять, почему нам нужно «упаковать» последовательность. Я понимаю, почему нам нужно их «прокладывать», но почему это pack_padded_sequenceнеобходимо? Приветствуются любые объяснения высокого уровня!

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

В numpy мы используем ndarray.reshape()для изменения формы массива. Я заметил, что в pytorch люди используют torch.view(...)с той же целью, но в то же время есть и torch.reshape(...)существующий. Поэтому мне интересно, в чем разница между ними и когда я должен использовать любой из них?

93 pytorch 

Я просматривал этот пример языковой модели LSTM на github (ссылка) . Что он делает в целом, мне довольно ясно. Но я все еще пытаюсь понять, что contiguous()делает вызов , что происходит несколько раз в коде. Например, в строке 74/75 создаются последовательности ввода кода и цели LSTM. Данные (хранящиеся в ids) …

90 neural-network  deep-learning  lstm  pytorch  contiguous 

Я пытаюсь обновить / изменить параметры модели нейронной сети, а затем сделать прямой проход обновленной нейронной сети в граф вычислений (независимо от того, сколько изменений / обновлений мы делаем). Я попробовал эту идею, но всякий раз, когда я делаю это, 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 …

9 python  machine-learning  gpu  pytorch