Only load the module specified for gen-summary-md (#622)
Summary:
Change this script to load only the request model, by exact match.
Sometimes there are un-desirable matches, and several models have
a similar name. Also, avoid overhead of loading all modules when
using load_model_by_name API.
Pull Request resolved: https://github.com/pytorch/benchmark/pull/622
Reviewed By: xuzhao9
Differential Revision: D32967211
Pulled By: aaronenyeshi
fbshipit-source-id: 9a8dec27a3e90291f5d0b2979a03485834db939f