Handle CUDA OOM in stableness check (#1517)
Summary:
Handle exceptions in stableness check, and output yaml files by default.
Pull Request resolved: https://github.com/pytorch/benchmark/pull/1517
Reviewed By: weiwangmeta
Differential Revision: D44631713
Pulled By: xuzhao9
fbshipit-source-id: 4ab3eb5b604d20f38fbe5c1671b0d4b5879743b2
