pytorch
784ae23d - ns for fx: fix bug in weight extraction testing (#55431)

Commit
4 years ago
ns for fx: fix bug in weight extraction testing (#55431) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/55431 Fixes a bug in the test cases, returning early resulted in some tests not being run. Adds logic for `nni.LinearReLU`, which was unmasked by making the tests run Test Plan: ``` python test/test_quantization.py TestFXNumericSuiteCoreAPIs.test_extract_weights_mod ``` Imported from OSS Reviewed By: hx89 Differential Revision: D27622415 fbshipit-source-id: 79d9e3125e5d881d9d13645abbe4bd007a5e1d44
Author
Parents
Loading