Correctness Check - Return False instead of assert (#1283)
Summary:
Pull Request resolved: https://github.com/pytorch/benchmark/pull/1283
Return False instead of assert enables better error message.
Reviewed By: xuzhao9, anijain2305
Differential Revision: D40970285
fbshipit-source-id: 684f88760e655a807f8139253d92ccb7d2f8fe22
