CRNN datset and dataloader

Hi,

I’m trying to build a CRNN with a convolutional layer followed by LSTM. The inputs are sequences of correlation matrices ,How do I design the dataset class? Is there a way to use VIdeoFrameDataset, but set it to accept every frame? There’s only 100 and each one is important. Thank you