would be better if you comment on how you load your data,  if you are using standard data loaders of pytorch,  or writing your own,  what is your data(images,  videos,  text) etc. Basically if you give more info or better your code,  the awesome ppl here would be able to help you out