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,...
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...
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
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...
Bagaimana cara menginisialisasi bobot dan bias (misalnya, dengan inisialisasi He atau Xavier) dalam jaringan di 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...
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...
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...
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...
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...