Pertanyaan yang diberi tag «pytorch»

Saya bingung tentang metode view()dalam cuplikan kode berikut. 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 = …

205 python  memory  pytorch  torch  tensor 

Saya mencari cara alternatif untuk menyelamatkan model yang terlatih di PyTorch. Sejauh ini, saya telah menemukan dua alternatif. torch.save () untuk menyimpan model dan torch.load () untuk memuat model. model.state_dict () untuk menyimpan model yang terlatih dan model.load_state_dict () untuk memuat model yang disimpan. Saya telah menemukan diskusi ini di …

192 python  serialization  deep-learning  pytorch  tensor 

Saya ingin tahu apakah pytorchmenggunakan GPU saya. Ada kemungkinan untuk mendeteksi nvidia-smijika ada aktivitas dari GPU selama proses, tetapi saya ingin sesuatu yang ditulis dalam pythonnaskah. Apakah ada cara untuk melakukannya?

160 python  memory-management  gpu  nvidia  pytorch 

Apakah ada cara, saya dapat mencetak ringkasan model di PyTorch seperti model.summary()metode yang dilakukan di Keras sebagai berikut? 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, …

125 python  pytorch 

Bagaimana cara menginisialisasi bobot dan bias (misalnya, dengan inisialisasi He atau Xavier) dalam jaringan di PyTorch?

117 python  neural-network  deep-learning  pytorch 

Saya membaca dokumentasi PyTorch dan menemukan contoh di mana mereka menulis gradients = torch.FloatTensor([0.1, 1.0, 0.0001]) y.backward(gradients) print(x.grad) di mana x adalah variabel awal, dari mana y dibangun (vektor 3). Pertanyaannya adalah, apa argumen 0.1, 1.0 dan 0.0001 dari tensor gradien? Dokumentasinya tidak terlalu jelas tentang itu.

112 neural-network  gradient  pytorch  torch  gradient-descent 

Saya mencoba untuk mereplikasi Bagaimana menggunakan pengepakan untuk input urutan variabel-panjang untuk rnn tapi saya rasa saya harus terlebih dahulu memahami mengapa kita perlu "mengemas" urutan. Saya mengerti mengapa kita perlu "mengemas" mereka tetapi mengapa "pengepakan" (melalui pack_padded_sequence) perlu? Penjelasan tingkat tinggi apa pun akan dihargai!

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

Saya telah melalui contoh model bahasa LSTM ini di github (tautan) . Apa yang dilakukannya secara umum cukup jelas bagi saya. Tapi saya masih berjuang untuk memahami apa contiguous()fungsi panggilan , yang terjadi beberapa kali dalam kode. Misalnya pada baris 74/75 dari input kode dan urutan target LSTM dibuat. Data …

90 neural-network  deep-learning  lstm  pytorch  contiguous 

Saya mencoba memperbarui / mengubah parameter model neural net dan kemudian meneruskan pass dari neural net yang diperbarui berada dalam grafik perhitungan (tidak peduli berapa banyak perubahan / pembaruan yang kami lakukan). Saya mencoba ide ini tetapi setiap kali saya melakukannya pytorch menetapkan tensor saya yang diperbarui (di dalam model) …

10 python  machine-learning  neural-network  pytorch 

Saya mencoba untuk melatih CNN berikut sebagai berikut, tetapi saya terus mendapatkan kesalahan yang sama mengenai .cuda () dan saya tidak yakin bagaimana cara memperbaikinya. Berikut adalah sebagian dari kode saya sejauh ini. import matplotlib.pyplot as plt import numpy as np import torch from torch import nn from torch import …

9 python  machine-learning  gpu  pytorch