expose example_input argument in setup_qat_model() - Githubissues

facebookresearch / d2go

D2Go is a toolkit for efficient deep learning

Apache License 2.0

837 stars 200 forks source link

expose example_input argument in setup_qat_model() #647

Closed stephenyan1231 closed 8 months ago

stephenyan1231 commented 8 months ago

Summary: Major changes

example_input argument in prepare_fake_quant_model() is useful in certain cases. For example, in Argos model custom_prepare_fx() method under FX graph + QAT setup (D52760682), it is used to prepare example inputs to individual sub-modules by running one forward pass and bookkeeping the inputs to individual sub-modules. Therefore, we export argument example_input in setup_qat_model() function.
For QAT model, currently we assert # of state dict keys (excluding observers) should be equal to # of state dict keys in the original model. However, when the assertion fails, it does not log useful information for debugging. We make changes to report what are the unique keys in each state dict.

Differential Revision: D52760688

facebook-github-bot commented 8 months ago

This pull request was exported from Phabricator. Differential Revision: D52760688

facebook-github-bot commented 8 months ago

This pull request was exported from Phabricator. Differential Revision: D52760688

facebook-github-bot commented 8 months ago

This pull request was exported from Phabricator. Differential Revision: D52760688

facebook-github-bot commented 8 months ago

This pull request was exported from Phabricator. Differential Revision: D52760688

facebook-github-bot commented 8 months ago

This pull request was exported from Phabricator. Differential Revision: D52760688

facebook-github-bot commented 8 months ago

This pull request was exported from Phabricator. Differential Revision: D52760688

facebook-github-bot commented 8 months ago

This pull request was exported from Phabricator. Differential Revision: D52760688

facebook-github-bot commented 8 months ago

This pull request was exported from Phabricator. Differential Revision: D52760688

facebook-github-bot commented 8 months ago

This pull request has been merged in facebookresearch/d2go@3c6f71b4ce8a27da8d39e9746f1f4b75367b2693.