-
In the paper replicating code when I write the following:
```
# For this notebook to run with updated APIs, we need torch 1.12+ and torchvision 0.13+
try:
import torch
import torchvision…
-
from torchinfo import summary
summary(model=pretrained_vit,
input_size=(32, 3, 128, 128),
col_names = ['input_size', 'output_size', 'num_params', 'trainable'],
col_width…
-
Has anyone tried torchinfo with BEVFusion? I tried it, but it reported that "TypeError: Model contains a layer with an unsupported input or output type: , type: "
-
```python
# Continue with regular imports
import matplotlib.pyplot as plt
import torch
import torchvision
from torch import nn
from torchvision import transforms
# Try to get torchinfo, ins…
-
**Describe the bug**
When the linear layer has a multidimensional input and output (shape with 3 dimensions or more) the computed multiple adds will be incorrect.
**To Reproduce**
Add lin…
-
**Describe the bug**
When trying to train (fine-tune) a Mistral model, there is an error when not using quantization.
**To Reproduce**
Steps to reproduce the behavior:
1. Install ludwig as usu…
-
hello. when i use torch summary. it reports some issues about:
File “F:\Anaconda3\lib\site-packages\torchsummary\torchsummary.py”, line 23, in
[-1] + list(o.size())[1:] for o in output
AttributeE…
-
**Describe the bug**
torchinfo.py", line 448, in traverse_input_data
result = aggregate(
TypeError: unsup…
-
**Describe the bug**
For ViT, the returned total mult-adds from torchinfo.summary is much smaller than that reported in other websites.
**To Reproduce**
Code snippet:
```python
from torchin…
-
### 🚀 Feature
same as the `input_data` as it in https://github.com/TylerYep/torchinfo
### Motivation & pitch
for seq2seq task, some times `xs` `xs_len` are related, so I want genertaed by myself.
…