search

mzolfaghari / ECO-efficient-video-understanding

Code and models of paper " ECO: Efficient Convolutional Network for Online Video Understanding", ECCV 2018
MIT License
437 stars 96 forks source link

2D tensor changed to 3D tensor, runtime error #45

Closed ZealACMer closed 5 years ago

ZealACMer commented 5 years ago

I tried to change 2D tensor into 3D tensor, so it can be feed into 3D Conv. The code are shown as follows: y = self.con1x1(x) # 2D tensor y = y.view((-1, 96, 16) + y.size()[2:]) #2D tensor changed to 3D tensor

error: image Thank you so much for you kind help.

mzolfaghari commented 5 years ago

@ZealACMer Are you sure that the number of samples is 16? 3161088 is not equal to batch96162828. I think it's better to check the shape of "y" after 2D tensor and figure out the problem. Let me know if you still didn't find the problem.