Am I totally missing the mark? Is there already some sort of pytorch item that does something like this but more efficiently? My fear is that every time a batch finishes the model doesn’t “remember” the last batches training because this is not a traditional dataset. Is this fear real?