I am receiving out of memory error after successfully finished first epoch with training and evaluation stage finished. How it is possible that I get out of memory after first epoch? I use images of the same size, same batch_size. Is there some sort of garbage collection?