Feature Extraction

[gkv91]

Hi,
p3d = P3D199(pretrained=True,num_classes=400,modality='RGB')
modules=list(p3d.children())[:-2]
nw_p3d=nn.Sequential(*modules)
data=torch.autograd.Variable(torch.rand(10,3,16,224,224))
p3d_out=nw_p3d(data)

I wrote the above code to extract video features. But it gave the error, ValueError: Expected 4D tensor as input, got 5D tensor instead. What will be the reason?
Thanks

[AndyHon]

Have you solved this problem?how did you solve it?