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

Fix loading weights in n-dim dense -> 1x1 conv #821

Merged
merged 2 commits into from Jun 28, 2023

Conversation

vloncar
Copy link
Contributor

@vloncar vloncar commented Jun 23, 2023

Description

In a change from using the readder to storing weights as attributes we missed the case of creating PointwiseConv nodes from n-dimensional Dense layer. This fixes it and adds a test for the future.

Type of change

  • Bug fix (non-breaking change that fixes an issue)

Tests

Added a test in test_multi_dense.py

Checklist

  • I have read the guidelines for contributing.
  • I have commented my code, particularly in hard-to-understand areas.
  • I have made corresponding changes to the documentation.
  • My changes generate no new warnings.
  • I have installed and run pre-commit on the files I edited or added.
  • I have added tests that prove my fix is effective or that my feature works.

@vloncar vloncar requested a review from thaarres June 23, 2023 15:13
@vloncar vloncar added the please test Trigger testing by creating local PR branch label Jun 23, 2023
Copy link
Contributor

@thaarres thaarres left a comment

Choose a reason for hiding this comment

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

This fixes an issue we were having when attempting to convert a DeepSets model with a stack of 2D dense layers

@thaarres thaarres merged commit ff756b8 into fastmachinelearning:main Jun 28, 2023
5 checks passed
calad0i pushed a commit to calad0i/hls4ml that referenced this pull request Jul 1, 2023
Fix loading weights in n-dim dense -> 1x1 conv
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
please test Trigger testing by creating local PR branch
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

2 participants