-
![bdab56019de5cdf38af366a483e3b32](https://user-images.githubusercontent.com/25839884/200819469-3cfdcaab-49d4-4e73-8108-bf819ca6892b.png)
-
**Describe the bug**
my code:
```
import torch
from PIL import Image
from thop import profile, clever_format
from torchinfo import summary
import cn_clip.clip as clip
from cn_clip.clip import…
-
Hello,
I managed without problem to run `torchinfo.summary` on `YOLOv5n` that I loaded from the torch hub.
```pycon
>>> torchinfo.summary(torch.hub.load('ultralytics/yolov5', 'yolov5n'), in…
-
I'd like to know if there is a way to get a model summary (preferably via torchinfo/troch-summary)? What I tried so far to get a summary for Mask-RCNN:
`from torchinfo import summary`
`summary(mod…
-
**Describe the bug**
I'm using torchinfo summary to see the number of parameters of UNet in Stable-Diffusion-2, but find it unmatch what is reported publicly.
**To Reproduce**
```python
from di…
-
**Describe the bug**
torchinfo.summary should not change the device of model
**To Reproduce**
model = model.cpu()
torchinfo.summary(model)
why model was changed to cuda after calling torchinfo…
gitxw updated
1 month ago
-
![image](https://github.com/wavefrontshaping/complexPyTorch/assets/136748433/2339e125-dac6-4b9d-9244-0d2eeeda1000)
-
my code:
```
import torch
from PIL import Image
from thop import profile, clever_format
from torchinfo import summary
import cn_clip.clip as clip
from cn_clip.clip import load_from_name, avail…
-
Hello,
I need detailed information about the model, and I would like to use torchinfo's summary function. Could you advise on where I should incorporate the following code?
`from torchinfo import su…
-
### System Info
- `transformers` version: 4.29.2
- Platform: macOS-14.2.1-arm64-arm-64bit
- Python version: 3.10.14
- Huggingface_hub version: 0.23.0
- Safetensors version: 0.4.3
- PyTorch versi…