Below is the link to the author’s i3d network. In their case they frame-wise multi-label classification.
[https://github.com/piergiaj/pytorch-i3d 15](http://i3d 2 Network for charades dataset)
I’m using Visual-Tactile dataset and



 ptrblck:

So the output shape would correspond to [batch_size, nb_classes, frames] ?


I3D is designed on kinetics dataset and I didn’t change default architecture from the above link having file “pytorch_i3d.py”.



 ptrblck:

I’m not that familiar with the i3d model, but I assume the temporal (and spatial) dimensions were reduced somehow?


I’m also new to this. But according to the author input frames to the network is 64. So each video i have converted to 64 frames. because I do not have to do multilabel classification.



 ptrblck:

The current output format would correspond to a frame-wise multi-label classification.


Probably I need to change the final layer since i don’t want multi-label classification.



 ptrblck:

In that case, you could use nn.BCEWithLogitsLoss (or nn.BCELoss + sigmoid) with a since output channel. Alternatively you could keep the two output channels and use nn.CrossEntropyLoss .


I still have the issue with dimension because it is clear that target and output are not of the same shape or as expected input to the loss function.