I’m a beginner to pytorch and implementing i3d network for binary classification. I have RGB video (64 frames simultaneously) input to the network and each video have a single label which is 0 (failure) or 1 (success).
I kept my batch size to 5 just to check if my network or code is working or not. (I would call it a debug run)



 ptrblck:

Also, it seems you are dealing with a multi-label classification, i.e. each sample might contain none, one, or both classes. Is this correct?


Each class have a single label. So I guess I should change network output from 2 to 1.