Skip to content
New issue

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

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

Already on GitHub? Sign in to your account

enh: linear growing module dimensions #54

Conversation

stephane-rivaud
Copy link
Collaborator

@stephane-rivaud stephane-rivaud commented Jan 17, 2025

Modify LinearGrowingModule such that is can accept input tensor with dimensions [*, dim] instead of simply [batchsize, dim].

WARNING: The tests associated to LinearGrowingModule might need to be adapted for robustness.

…appens if running mlp_run.py script on 'sin' dataset.
…dimensions [*, dim] instead of simply [batchsize, dim].
…sions' into enh-linear_growing_module-dimensions
Copy link
Collaborator

@sylvchev sylvchev left a comment

Choose a reason for hiding this comment

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

Ok, this implementation is ok for me. Could you fix the tests and add a line in whatsnew?

@sylvchev
Copy link
Collaborator

Fixes #51

@sylvchev sylvchev added the enhancement New feature or request label Jan 20, 2025
Copy link
Collaborator

@TheoRudkiewicz TheoRudkiewicz left a comment

Choose a reason for hiding this comment

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

The current changes seems good. I think we need to do the same work in the Addition part.

Copy link

codecov bot commented Feb 13, 2025

Codecov Report

All modified and coverable lines are covered by tests ✅

Flag Coverage Δ
unittests 68.54% <ø> (ø)

Flags with carried forward coverage won't be shown. Click here to find out more.

Files with missing lines Coverage Δ
src/gromo/linear_growing_module.py 58.36% <ø> (ø)

Copy link
Collaborator

@sylvchev sylvchev left a comment

Choose a reason for hiding this comment

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

Ok, good to go, thanks @stephane-rivaud !

@stephane-rivaud stephane-rivaud merged commit 143c524 into growingnet:main Feb 13, 2025
6 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancement New feature or request
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants