-
### 🐛 Describe the bug
Hi!
Following snippet produces errors:
```py3
import torch
print(torch.__version__) # Tried with 2.1.0+cu121 (latest) and 2.0.1 (default in google colab). All fails
clas…
-
### Bug description
Following the most basic example for MNIST with a multilayer model. Does not run properly on my Mac M1 laptop.
```python
from typing import Any
import lightning.pytorch as pl…
-
Hi everytime you run summary() a UserWarning pops up.
This is a warning message that is telling you that the use of TypedStorage in your code is deprecated and will be removed in the future. Instea…
-
Hi,
I am trying to summarise a model from HuggingFace Hub
```python
from transformers import T5ForConditionalGeneration, T5Tokenizer, T5Config
config = T5Config.from_pretrained('t5-large')
…
-
### Search before asking
- [X] I have searched the YOLOv5 [issues](https://github.com/ultralytics/yolov5/issues) and [discussions](https://github.com/ultralytics/yolov5/discussions) and found no simi…
-
Hi,
I recently implemented model parallelism. With it, however, the code fails in
torchinfo.summary( self.model, input_data=[batch_data])
with "RuntimeError: Expected all tensors to be on th…
-
**Describe the bug**
`torchinfo.summary()` fails on a model that returns numpy ndarray instead of tensors.
**To Reproduce**
first install https://mmpose.readthedocs.io
```
from pathlib im…
-
### 是否已有关于该错误的issue或讨论? | Is there an existing issue / discussion for this?
- [X] 我已经搜索过已有的issues和讨论 | I have searched the existing issues / discussions
### 该问题是否在FAQ中有解答? | Is there an existing…
-
Hi everyone,
I am getting the following error while loading a model trained with brevitas
```python
Traceback (most recent call last):
File "tools/test_widerface.py", line 258, in
main()
…
-
Versions
```
torch.__version__
'1.13.1+cu117'
transformers.__version__
'4.20.0'
torchinfo.__version__
'1.7.2'
```
To Reproduce
```
import torch
from transformers import GPT2Model, GP…