Skip to content
New issue

Have a question about this project? # for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “#”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? # to your account

Remove reset of dataloader in len method of ArrayLoader #123

Merged

Conversation

oliverholworthy
Copy link
Member

@oliverholworthy oliverholworthy commented Apr 4, 2023

Following #111 removing the reset of the dataloader (call to the stop method) inside the __len__ method of ArrayLoader.

This loader reset in the len method caused some infinte loops in Transformers4Rec where we call len on the dataloader during it's iteration in an evaluation_loop method of the trainer.

@oliverholworthy oliverholworthy added this to the Merlin 23.04 milestone Apr 4, 2023
@oliverholworthy oliverholworthy self-assigned this Apr 4, 2023
This reset of the loader was only required for the Keras Sequence dataloader.
@oliverholworthy oliverholworthy marked this pull request as ready for review April 4, 2023 14:15
@oliverholworthy oliverholworthy merged commit fa716be into NVIDIA-Merlin:main Apr 4, 2023
@oliverholworthy oliverholworthy deleted the remove-len-reset branch April 4, 2023 14:44
# for free to join this conversation on GitHub. Already have an account? # to comment
Labels
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants