What I meant was I need a way to keep track of the video frames from a certain video when I load in the dataset instead of each of them being a seperate entitiy.
Alternately, is there something like the keras SlidingFrameGenerator in pytorch?