-
### Bug description
`all_gather` function somehow produces negative values.
Here is the code snippet I have in `on_train_epoch_end` function:
```
if torch.any(self.query_labels < 0):
…
-
**Describe the bug**
```python
from torchvision.models import resnet18
from torchinfo import summary
model = resnet18()
a = summary(model, (1, 3, 256, 256))
```
In `python`, `ipython`, this…
-
### 🐛 Describe the bug
When using [generics for typing](https://peps.python.org/pep-0585/#reference-implementation) in Python 3.9, I run into this error
```
Traceback (most recent call last):
…
-
hello, could you offer me the detailed training environment? I accoured the problems in this environment and use the docker:
Package Version
---------------------- -------------------…
-
**Describe the bug**
For CPU usage, torchinfo.summary works well.
For models on GPU, if the model were not on GPU 0 (e.g. on GPU 5), torchinfo.summary will move it to GPU 0.
**To Reproduce**
Ste…
-
### 🐛 Describe the bug
I performed a simple **classification** inference on a sample image using ResNet-50. I used both the **`IMAGENET1K_V1`** and **`IMAGENET1K_V2`** versions of the model weights.…
-
### 🐛 Describe the bug
I'm trying to move a state_dict from python to c++ per the following code and pickle_load() fails with an error requesting that I report the bug. I've attached the zipped versi…
-
### Bug description
The class Trainer says that setting to ``"warn"`` to use deterministic algorithms whenever possible, throwing warnings on operations that don't support deterministic mode (require…
-
### Bug description
I tried set precision in the config yaml for lightning-cli, but failed. The following is my yaml file.
```python
# lightning.pytorch==2.0.6 jsonargparse==4.22.1
seed_everythi…
-
### 🐛 Describe the bug
Attempting to jit trace the ViT model in eval mode raises an exception `TracingCheckError: Tracing failed sanity checks! ERROR: Graphs differed across invocations!`
Minimal …