-
I have used this library to calculate jacobians for the [Neural tangent kernel](https://en.wikipedia.org/wiki/Neural_tangent_kernel) study.
It's fast and reliable and I'm just thinking to **extend th…
-
Hi,
I find torchinfo seems omit the MAccs of some operations like `torch.mul` and `torch.matmul`. Is it some feature-to-be-added, or we do not need to count these operations when calculating MAccs?…
-
Trying to summarize a [CLIP](https://github.com/openai/CLIP) model results in an output where there are no values listed in the `Output Shape` column:
```
===========================================…
-
Hey,
This is a really great tool to visualize the model. However, I was trying to see how my decoder is working in the VAE and the input to the VAE is the latent space (dim = `(2,2)`). However, whe…
-
**Is your feature request related to a problem? Please describe.**
Every time I want to put the specific structure of the model I use into a thesis, I will use torchinfo to print it, and make a latex…
-
**Describe the bug**
I have built a cnn architecture using conv1d and activation layers for a regression task at hand. Now when I am trying to print the summary of the architecture it is showing me t…
-
您好,请问您用的什么工具分各层计算参数量呢?可以分层计算计算量,如FLOPs吗?
-
Imagine I have a module like this:
```
class Model(nn.Module):
def __init__(self, generator, discriminator):
super(Model, self).__init__()
# Define Generator
self.g…
-
Hi there
Your idea of using the transformer for event-based depth estimation is great.
I'm working on Python 3.9 and CUDA 11.8 and trying to update the environment.
The first is the summery fu…
-
Hi there.
I was trying to inspect mmoe model from [mmoe](https://github.com/huangjunheng/recommendation_model), which has nn.PrameterList.
```python
class Expert(nn.Module):
def __init__(…