DRAGNLabs / 301r_retnet

2 stars 1 forks source link

Fix iterator bug #47

Closed JayOrten closed 6 months ago

JayOrten commented 6 months ago

This fixes a bug where: when creating an iterator across the rows of the data, for any dataloader, the code would reassign the object data variable to the generator, which would throw an error when the iter function was called again. Instead, we assign the generator to a local variable.

jtappen1 commented 6 months ago

Looks good to me! I am testing it now.