Skip to content

Remove obsolete parameter mapping between pytorch and keras

Javier Duarte requested to merge github/fork/JanFSchulte/batchNormFix into main

Created by: JanFSchulte

The pytorch parser still uses a mapping between tensor names from keras to pytorch which has become obsolete. In its current form, it breaks BatchNorm layers. This PR simply removes this functionality, restoring BatchNorm functionality and removing clutter. A test for BatchNorm in pytorch is added to prevent further breaking of this functionality

Type of change

For a new feature or function, please create an issue first to discuss it with us before submitting a pull request.

Note: Please delete options that are not relevant.

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

Tests

Ran pytests to verify that nothing gets broken, verified that BatchNorm layers parse successfully after the fix.

Checklist

  • I have read the guidelines for contributing.
  • 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