我有一个在 python 中训练的模型,使用 跟踪torch.jit.trace
,并使用torch::jit::load
.
有没有办法访问最后一层以提取模型所需输出深度的值(例如,如果它是从 16 -> 2 开始的 Conv2D 层,我想为深度 [b,d- >2,x,y] 的 2)?
我有一个在 python 中训练的模型,使用 跟踪torch.jit.trace
,并使用torch::jit::load
.
有没有办法访问最后一层以提取模型所需输出深度的值(例如,如果它是从 16 -> 2 开始的 Conv2D 层,我想为深度 [b,d- >2,x,y] 的 2)?