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

LSTNet Implementation #173

Merged

Conversation

Added conv1 custom args and placeholder for GRU model
WIP, added layer args. GRU still needs to be implemented
Debugging reshape sizing with Linear ODE notebook.
Fixed imports for new module.
@stefanradev93 stefanradev93 marked this pull request as ready for review June 11, 2024 12:36
stefanradev93 and others added 6 commits June 11, 2024 14:39
Condensed layers to sequential model.
Removed reshaping layers as they are no longer necessary.
Moved GRU and Skip GRU to skip_gru.py.
Implemented SkipGRU.
Added SkipGRU and ResNet to LSTNet.
Fixed bug that attempted to reshape over None dimension when training offline.
Updated build() for both modules.
Condensed comments in lstnet.
Added new kwargs retrieval method in lstnet.py and skip_gru.py
Tuned default arguments for both LSTNet and SkipGRU.
Condensed arithmetic in SkipGRU.
@Chase-Grajeda Chase-Grajeda changed the title LSTNet Implementation (WIP) LSTNet Implementation Jun 13, 2024
@Chase-Grajeda Chase-Grajeda self-assigned this Jun 13, 2024
@Chase-Grajeda Chase-Grajeda added the feature New feature or request label Jun 13, 2024
Removed gru.py since keras has it's own implementation
Copy link
Contributor

@stefanradev93 stefanradev93 left a comment

Choose a reason for hiding this comment

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

Well done! Will do some efficiency edits on the skip gru to avoid the for loop and push these for a performance check.

@stefanradev93 stefanradev93 merged commit 2f87322 into bayesflow-org:streamlined-backend Jun 13, 2024
12 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
feature New feature or request
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants