Add cpu_peak_mem and gpu_peak_mem to the metrics
Summary:
Use the new correctness/accuracy checking interface to run the model.
Add new metrics including cpu_peak_mem and gpu_peak_mem.
Reviewed By: davidberard98
Differential Revision: D46395827
fbshipit-source-id: f33535b9a6f8830b533734f4618733e478bd0ab9