Hi,
My code is pretty standard, I have a training and testing loop over train and test sets. I’m using a dataloader object for both. The problem I’m having is that when the training loop is over, the train_loader doesn’t release its GPU memory. Therefore, when I start iterating over the test_loader (with same batch size) I get a memory error. Is there a way to tell a dataloader to relinquish its GPU memory in order to avoid such error ?
Many thanks,
Lucas