benchmark
e02e7ff0 - Check type of example inputs from model.get_module() (#1167)

Commit
3 years ago
Check type of example inputs from model.get_module() (#1167) Summary: The example test checker unpacks example inputs either as **kwargs, if the example inputs are a dict instance, or as *args by default. We should only unpack as *args if the example inputs are a tuple instance, otherwise raise an assert error by default. Unpacking non-tuple inputs as *args can lead to tricky dimension mismatch errors. Pull Request resolved: https://github.com/pytorch/benchmark/pull/1167 Reviewed By: xuzhao9 Differential Revision: D39354742 Pulled By: nmacchioni fbshipit-source-id: f6e2cd21fa9149a55e201ce516301dc0dce715aa
Author
Parents
Loading