Skip to content

Conversation

@oliverholworthy
Copy link
Contributor

@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
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants