-
### 🚀 The feature, motivation and pitch
A model summary functionality to print a string summary of the network/model, like the `tensorflow.Keras.Model.summary()` method. A model summary would als…
-
### 🐛 Describe the bug
`torch.abs`/`Tensor.abs` produces `NaN`s in `AbsBackward0` for small complex numbers.
```python
import torch
x = torch.tensor([0.0 + 0.0j, -1.0568e-23 + 0.0j], requir…
-
I use `torchinfo` to get a rough understanding of model sizes and how the "weight" is distributed across the model. Unfortunately, the latter information is not available directly, which is not conven…
-
**Is your feature request related to a problem? Please describe.**
I was facing the issue described [here](https://stackoverflow.com/questions/74609050). I wanted to train on CPU, although CUDA was a…
-
### 🐛 Describe the bug
Hi,
Related issues: https://github.com/pytorch/pytorch/issues/62712 https://github.com/pytorch/pytorch/issues/83974 @justinchuby @shubhambhokare1
This code runs fine:
…
-
### 🐛 Describe the bug
Code
```python
import torch
print(torch.arange(9).reshape(3, 3).to("cpu")[[1, 2], [2, 1]])
print(torch.arange(9).reshape(3, 3).to("mps")[[1, 2], [2, 1]])
```
Output
``…
-
hello,
I use craft from https://github.com/clovaai/CRAFT-pytorch,
the torchsummary output is:
![image](https://user-images.githubusercontent.com/12007227/133239645-759cbbaf-4758-4c99-9e12-afd8f9…
-
It would be very useful to have a way to specify the units (MB, GB, etc ) in which the number of parameters and MACS are reported. This could help quickly compare different architectures.
I think o…
-
### 🐛 Describe the bug
To Reproduce:
```
>>> import torchvision
>>> torchvision.transforms.InterpolationMode.NEAREST
>>> torchvision.transforms.InterpolationMode.NEAREST_EXACT
Traceback (most …
-
### Checklist
- [X] I added a descriptive title
- [X] I searched open reports and couldn't find a duplicate
--> CLOSED duplicates are #4484 #4481 #4509
### What happened?
Building a conda…