pytorch
5b93cdac - ns for fx: remove model_name from get_matching_activations API (#52926)

Commit
3 years ago
ns for fx: remove model_name from get_matching_activations API (#52926) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/52926 Model name is already stored in the Loggers in the prepare call. Removing the need to specify it again in the extract activations functions, to simplify things. Test Plan: ``` python test/test_quantization.py TestFXNumericSuiteCoreAPIs ``` Imported from OSS Reviewed By: raghuramank100 Differential Revision: D26693473 fbshipit-source-id: 52511cacc16f79fa09c78ccde78e7f439f4b315c
Author
Parents
Loading