Bakaloader -

for batch in loader.stream(): model.training_step(batch)

for batch in loader.stream(): model.training_step(batch)