Hi, thanks for your reply. I think I don’t require the data loader. I precomputed the resnet features and saved in a pickle file (of size around 50gb). My model just uses these features. The problem is I am getting out of memory error after some epochs.
Thanks