Fix issues encountered when running models with lazy tensors (#697)
Summary:
Also addresses niter inconsistency https://github.com/pytorch/benchmark/issues/687
Pull Request resolved: https://github.com/pytorch/benchmark/pull/697
Reviewed By: xuzhao9
Differential Revision: D33660895
Pulled By: wconstab
fbshipit-source-id: d9dedbb5b51be016bf7144658168adfa90e485fe