-
Notifications
You must be signed in to change notification settings - Fork 4
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
enh: linear growing module dimensions #54
Conversation
…appens if running mlp_run.py script on 'sin' dataset.
…dimensions [*, dim] instead of simply [batchsize, dim].
…sions' into enh-linear_growing_module-dimensions
There was a problem hiding this 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?
Fixes #51 |
There was a problem hiding this 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.
Codecov ReportAll modified and coverable lines are covered by tests ✅
Flags with carried forward coverage won't be shown. Click here to find out more.
|
There was a problem hiding this 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 !
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.