Questions tagged [pytorch]

The tag has no usage guidance.

Filter by
Sorted by
Tagged with
0
votes
0answers
27 views

Binary classification with pytorch

I wrote a simple neural network binary classification algorithm using Pytorch. It uses the dataset from https://www.kaggle.com/pritsheta/heart-attack, which consists of a table with 300 rows and 14 ...
0
votes
1answer
20 views

Set min value of each row of a tensor to zero without using explicit loops

Here's the problem. Return a copy of x, where the minimum value along each row has been set to 0. For example, if x is: ...
0
votes
0answers
12 views

Pytorch Video Frames Dataset Implementation

I have a dataset of video frames in a folder with the directory structure looking like this: -parent -video_id -frame0 -frame1 -... My code, gathering ...
2
votes
1answer
62 views

Loss function in python is a bit cumbersome

This is my first post here. I seek to improve my way of writing Python and C++ code. I hope I can also contribute to others when I have increased my skill. Here is a python function I have for a ...
1
vote
0answers
33 views

resnet with pytorch

I was studying resnet and wanted to program it with pytorch I searched for some examples(github, google) but it was hard to understand the code completely So I programmed resnet myself and it works. ...
1
vote
0answers
62 views

Complex division for imaginary part

I seek a fast implementation of (x / y).imag, where x, y are complex 2D arrays (PyTorch tensors already on GPU). My approach is ...
7
votes
1answer
109 views

A .py utility file for neural network learing rate policies

I've created a .py utility file, which specifies the learning rate policy for the neural network in PyTorch. The program in prior reads a ...
2
votes
0answers
91 views

Computing inverse of modified matrix using Sherman-Morrison formula

I have the following implementation that takes in symmetric matrix W and returns matrix C ...
3
votes
1answer
88 views

increase efficiency of loops and element-wise operations in PyTorch implementation

For any input symmetric matrix with zero diagonals W, I have the following implementation in PyTorch. I was wondering if the ...
2
votes
0answers
22 views

Enhancing performance using DataParallel

I have written the following code to practice parallelizing a PyTorch code on GPUs: ...
3
votes
0answers
56 views

Batch normalization code optimization?

I am trying to implement Deep Quaternion Networks. I was able to implement the batch normalization technique. But it requires a lot of GPU memory. Is there any way I can optimize the code provided ...
1
vote
1answer
1k views

Loops in PyTorch Implementation

I'm trying to implement a regularization term for the loss function of a neural network. ...
4
votes
2answers
856 views

PyTorch Vectorized Implementation for Thresholding and Computing Jaccard Index

I have been trying to optimize a code snippet which finds the optimal threshold value in a n_patch * 256 * 256 probability map to get the highest Jaccard index ...
2
votes
0answers
83 views

GRU-Autoencoder training

I'm training a GRU auto-encoder and my current code is very slow.I believe it's predict_captions function that takes most the time. Any suggestion to optimize this code? ...
4
votes
1answer
2k views

PyTorch Unit-testing in Python

I'm new to PyTorch and I'm writing a unit test for an activation function I'm making. I plan to test against a reference implementation for this function. I want to approach this in a test-driven way,...
1
vote
0answers
24 views

Splitting torch tensor by flags

Working with PyTorch tensors, I need to split the batch of items by its flags, so the items in x_batch_one and x_batch_two are ...
1
vote
0answers
102 views

Posterior collapses in an RNN variational auto-encoder - PyTorch

I am trying to implement and train an RNN variational auto-encoder as the one explained in "Generating Sentences from a Continuous Space". Although I apply their proposed techniques to mitigate ...
18
votes
2answers
10k views

Intersection over Union for rotated rectangles

Problem Statement I am trying to find the intersection over union (IoU) metric for one to several rotated rectangles compared to many different rotated rectangles. Here are some images to help ...