Skip to content

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

Javier Duarte requested to merge github/fork/vloncar/multi_dense_fix into main

Created by: vloncar

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.

Merge request reports

Loading