-
https://github.com/TylerYep/torchinfo/blob/73ed5687acfd6199b77fa1dcb65aa54762c1b720/torchinfo/torchinfo.py#L501
```
>>> get_total_memory_used(["abc", "def"])
Traceback (most recent call last):
…
-
Hello -
the frozen versions of various packages currently shown in this project's `requirements.txt` cannot be installed in Python 3.12 environments. To proceed with the installation, versions for…
-
### 🚀 Describe the improvement or the new tutorial
This tutorial will help users understand how to transition from convolutional and pooling layers to linear layers in their models.
Learning objec…
-
in order to solve
ResolvePackageNotFound:
- pytorch::torchdata=0.3.0
- pytorch::torchtext=0.12.0
- pytorch::torchvision=0.12.0
I changed my python to version 3.9,
in order to inst…
-
when I use torchinfo with model taking dictionary as input
summary(self.model, input_data = [data_dict])
I meet a bug :
TypeError (note: full exception trace is shown but execution is paused …
-
Since this project hasn't been worked on in a few months, I decided to do a rewrite of the fixing a lot of the changes.
The changes are available on my fork and is installable using:
`pip install …
-
**Describe the bug**
I have a custom TextVectorization layer, it doesn't use any `nn.` layers, it's just a dictionary of words that is used to fill in a `torch.zeros` vector. I want it to be baked in…
-
At [06. PyTorch Transfer Learning](https://www.learnpytorch.io/06_pytorch_transfer_learning/)
It needs to download efficientnetB0 model from https://download.pytorch.org/models/efficientnet_b0_rwig…
-
根据 #166 中提到的打印模型权重信息的需求,计划很快上线相关功能
## 👾 What does the proposed API look like
```python
import swanlab
import torchvision
from torchvision.models import ResNet50_Weights
# load pretrain…
-
Hi,
I came across the notion of [static/dynamic flow](https://pytorch.org/docs/stable/fx.html#limitations-of-symbolic-tracing) on PyTorch's doc.
I realize that dynamic flow (that is when module ca…