Câu hỏi được gắn thẻ «pytorch»

Tôi bối rối về phương thức view()trong đoạn mã sau. 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))) …

205 python  memory  pytorch  torch  tensor 

Tôi đang tìm cách thay thế để cứu một mô hình được đào tạo ở PyTorch. Cho đến nay, tôi đã tìm thấy hai lựa chọn thay thế. Torch.save () để lưu mô hình và Torch.load () để tải mô hình. model.state_dict () để lưu mô hình đã được đào …

191 python  serialization  deep-learning  pytorch  tensor 

Tôi muốn biết nếu pytorchđang sử dụng GPU của tôi. Có thể phát hiện nvidia-sminếu có bất kỳ hoạt động nào từ GPU trong quá trình, nhưng tôi muốn một cái gì đó được viết bằng một pythontập lệnh. Có cách nào để làm vậy không?

160 python  memory-management  gpu  nvidia  pytorch 

Có cách nào, tôi có thể in tóm tắt của một mô hình trong PyTorch như model.summary()phương pháp làm trong Keras như sau không? Model Summary: ____________________________________________________________________________________________________ Layer (type) Output Shape Param # Connected to ==================================================================================================== input_1 (InputLayer) (None, 1, 15, 27) 0 ____________________________________________________________________________________________________ convolution2d_1 (Convolution2D) (None, 8, 15, 27) …

125 python  pytorch 

Tôi đang đọc tài liệu của PyTorch và tìm thấy một ví dụ nơi họ viết gradients = torch.FloatTensor([0.1, 1.0, 0.0001]) y.backward(gradients) print(x.grad) trong đó x là một biến ban đầu, từ đó y được xây dựng (một vectơ 3). Câu hỏi là, các đối số 0,1, 1,0 và 0,0001 …

112 neural-network  gradient  pytorch  torch  gradient-descent 

Phương thức zero_grad()cần được gọi trong quá trình đào tạo. Nhưng tài liệu không hữu ích lắm | zero_grad(self) | Sets gradients of all model parameters to zero. Tại sao chúng ta cần gọi phương thức này?

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

Làm thế nào để khởi tạo các trọng số và độ lệch (ví dụ: với khởi tạo He hoặc Xavier) trong một mạng trong PyTorch?

106 python  neural-network  deep-learning  pytorch 

Tôi đang cố gắng sao chép Cách sử dụng đóng gói cho các đầu vào chuỗi có độ dài thay đổi cho rnn nhưng tôi đoán trước tiên tôi cần hiểu tại sao chúng ta cần "đóng gói" chuỗi. Tôi hiểu tại sao chúng ta cần "độn" chúng nhưng tại …

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

Tôi đã xem qua ví dụ này về mô hình ngôn ngữ LSTM trên github (liên kết) . Những gì nó làm nói chung là khá rõ ràng đối với tôi. Nhưng tôi vẫn đang đấu tranh để hiểu những gì gọi contiguous()thực hiện, điều này xảy ra nhiều lần …

90 neural-network  deep-learning  lstm  pytorch  contiguous 

Trong numpy, chúng tôi sử dụng ndarray.reshape()để định hình lại một mảng. Tôi nhận thấy rằng trong pytorch, mọi người sử dụng torch.view(...)cho cùng một mục đích, nhưng đồng thời, cũng có một torch.reshape(...)tồn tại. Vì vậy, tôi tự hỏi sự khác biệt giữa chúng là gì và khi nào …

83 pytorch 

Tôi đang cố gắng cập nhật / thay đổi các tham số của mô hình mạng nơ ron và sau đó có đường truyền chuyển tiếp của mạng nơ ron cập nhật nằm trong biểu đồ tính toán (cho dù chúng tôi có thực hiện bao nhiêu thay đổi / …

10 python  machine-learning  neural-network  pytorch 

Tôi đang cố gắng đào tạo CNN sau như sau, nhưng tôi vẫn gặp lỗi tương tự liên quan đến .cuda () và tôi không chắc cách khắc phục. Đây là một đoạn mã của tôi cho đến nay. import matplotlib.pyplot as plt import numpy as np import torch from …

9 python  machine-learning  gpu  pytorch 

Tôi sử dụng Tensorflow, nhưng tôi đang viết tài liệu cho người dùng thường sẽ thay đổi theo các khung học tập sâu . Khi làm việc với các bộ dữ liệu không phù hợp với hệ thống tệp cục bộ (TB +), tôi lấy mẫu dữ liệu từ kho …

8 python  pytorch 

Tôi đang làm CNN với Pytorch cho một nhiệm vụ, nhưng nó sẽ không học và cải thiện độ chính xác. Tôi đã tạo một phiên bản làm việc với bộ dữ liệu MNIST để tôi có thể đăng nó ở đây. Tôi chỉ đang tìm kiếm một câu trả …

8 python  conv-neural-network  pytorch 

Có một mẹo nổi tiếng trong kiến ​​trúc u-net là sử dụng bản đồ trọng lượng tùy chỉnh để tăng độ chính xác. Dưới đây là các chi tiết của nó- Bây giờ, bằng cách hỏi ở đây và tại nhiều nơi khác, tôi đã biết về 2 cách tiếp …

8 python  pytorch  image-segmentation  semantic-segmentation