如题，比较简单的模型，要看代码加测试几个输入形状吗，有没有比较方便的工具
• 请不要在回答技术问题时复制粘贴 AI 生成的内容
This topic created in 1262 days ago, the information mentioned may be changed or developed.
 |
albert0yyyyMay 26, 2023
1
随便输入一个，报个错。看看要多大的
|
 |
565656May 26, 2023
2
自己随便输入报错里不就有了
|
 |
HuelseMay 26, 2023
3
报个错，tensorboard
|
 |
connectsixboyMay 26, 2023
4
模型文件导入 Netron 看一下
|
 |
psycho9631May 26, 2023
5
torchinfo
|
 |
6
|
|
7
|
|
565656May 26, 2023
8
@lyang #6 快速的话基本没有，因为有时候像第一层是卷积或者 transformer 层的话你可能知道一些输入大小，但是运行到后面如果有像全连接层的话还是会影响到输入的大小，所以必须走一遍全过程，最简单的方法就是输入报错
|
 |
coldheartMay 26, 2023
9
torchsummary
输入一个正确纬度的 tensor ，可视化各层的形状和参数量 |
 |
thinkershareMay 26, 2023
11
没什么好办法，如果没有模型原代码，只能用数据走一遍，如果前向传播时候传递的参数非常复杂，你甚至可能没法尝试出结果。我一般分析模型也是用 netron, 但是 ONNIX 对 pytroch 的支持还是有问题。
|
 |
ChekkibrkiMay 26, 2023
12
torchinfo.summary(model,Tensor)
Tensor:输入形状[N,C,H,W],然后用这个函数就能打印出每层的形状。 |
|
connectsixboyMay 30, 2023
13
@lyang 正常模型文件导入 Netron 就行，不用输入测试，Netron 就是为了应对你这类问题存在的
|
Select Language