Skip to content

Conversation

@rcurtin
Copy link
Member

@rcurtin rcurtin commented Oct 7, 2024

I debugged mlpack/examples#239 and found that the LSTM layer did not properly copy all of its internal members, leading to an attempted way-too-large memory allocations in some cases.

The fix is pretty simple: inSize and outSize just need to be set correctly.

Thanks @CyberShadow for the report!

Copy link
Member

@shrit shrit left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

@CyberShadow thanks for reporting this

Copy link
Contributor

@github-actions github-actions bot left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Second approval provided automatically after 24 hours. 👍

@shrit shrit merged commit 94a6b48 into mlpack:master Oct 11, 2024
@rcurtin rcurtin deleted the lstm-copy-fix branch October 15, 2024 20:09
@rcurtin rcurtin mentioned this pull request Dec 3, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants