-
My code runs and display two images but the segmentation is just very very poor similar to issue #24, can you see if I am calling the routines incorrectly.
```
image_batch_tensor = tf.expand_dims…
-
## 🐛 Bug
If I create multiple threads, each owning its own built in pytorch module such as `torch::Linear`, I don't get thread safety related errors when running the networks concurrently. Howe…
-
In my code, I'm loading two DNN models. Model A is a normal DNN with fully-connected layers, and Model B is a Convolutional Neural Network similar to the one used in the MNIST example.
Individually,…
-
Hi,
I'm researching capsule neural networks for brain image segmentation. For my project, I should do 3D convolutions on tensors that have 6 dimensions:
input size: (N, P, C_in, D_in, H_in, W_in…
-
# [Roadmap]
## 1.6.1
* [x] [**`WikiCSDataset`**](https://pytorch-geometric.readthedocs.io/en/latest/modules/datasets.html#torch_geometric.datasets.WikiCS)
* [x] **`DeepGCN`**
* [x] [`GENConv…
-
Hi,
I'm trying to use torch.nn.DataParallel on lpips network,but then it gives me error
But when I modified the 100th line in the `stylegan2-pytorch.lpips.dist_model.py` from `self.net = torch.n…
-
Hi Dominik,
I just noticed that both images and segmentations are of type float64 after running any subfunction.
That might lead to memory issues during training and the batches take a lot of spac…
-
Hey. Thanks for your work first!
I'm just wondering if we really have any speedup with the zero_padding approach cause in this case we restore our layer to the same previous number of parameters, do…
-
## 🚀 Feature
support mainstream pruning techniques.
## Motivation
Recently, lots of new pruning algorithms are proposed, but the [current implementation](https://github.com/pytorch/pytorch/blob/4…
he-y updated
11 months ago
-
## 🚀 Feature
I would like to request the addition of support for Recurrent Neural Networks (RNNs) in the KFACOptimizer class. Currently, the KFACOptimizer class works for linear and 2D convolution …