-
When I use “torchsummary.summary(model)” in test.py. It give me TypeError:forward() missing 1 required positional argument: 'img_metas'. How can I solve this question? Thanks a lot!
cyy18 updated
2 years ago
-
Dear professor, sorry to bother you again. I encountered an error while training train-vm.py. How should I solve it? It seems like I need to add the shape of the input data.
![a18997bf27344288223b23b…
-
안녕하세요. 열심히 Pycharm 돌려보고 있는 참가자 입니다.
다름이 아니라 현재 Model이 잘 학습하고 있나 확인하고자
torchsummary 함수를 이용하려 하는데, 계속 같은 오류가 나와 문의하고자 합니다.
input.shape을 통해 데이터가 [4, 384, 384] 크기의 tensor 라는 것을 확인했고, 이를
s…
-
import torch
from vit import ViT
from torchsummary import summary
import timm
v = ViT(
image_size = 224,
patch_size = 16,
num_classes = 1000,
dim = 768,
depth = 12,
h…
-
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…
-
按要求安装了所需版本的软件,结果提示这个错误,请问是怎么回事呢?
-
The number of parameters reported for CapsNet seems to be coming from [torchsummary](https://github.com/sksq96/pytorch-summary).
However, torchsummary parameter count is inaccurate for custom modul…
Thity updated
11 months ago
-
Hi Xintao:
I try to get EDVR summary info, while it can not work on the DCN part~
have you ever tried to get related info before? Could you please give me some tips about it~
Thx
Jie
-
I wonder how to calculate FLOPs for a model, where standard methods involve using torchstat(model, input.shape) or torchsummary. However, in this project, there are many different inputs, which makes …
-
### 🐛 Describe the bug
I have a problem training a model with the BCE loss using mps device. The model uses a linear layer to predict a the sign of a real value number (two classes). When I train it …