Questions tagged [pytorch]
PyTorch is a deep learning framework that implements a dynamic computational graph, which allows you to change the way your neural network behaves on the fly and capable of performing backward automatic differentiation.
6 questions with no upvoted or accepted answers
3
votes
0
answers
78
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 ...
- 221
2
votes
0
answers
110
views
Higher order derivatives with functorch
I'm computing higher order derivatives using nested jacobian from pytorch/functorch.
$$f(x, y, z, \dots) : R^{m_x} \times R^{m_y}...
- 121
2
votes
0
answers
38
views
Enhancing performance using DataParallel
I have written the following code to practice parallelizing a PyTorch code on GPUs:
...
- 337
2
votes
0
answers
185
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?
...
- 169
1
vote
0
answers
181
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 ...
- 163
0
votes
0
answers
84
views
Convolution to reduce popcorn noise in black and white video
I have a black & white video as a tensor with the shape [Time, Width, Height] with popcorn noise and I would like to reduce the noise by naively convolving along the time dimension.
Using a ...
- 1,617