benchmark
a0996581 - Move torchbench model configuration into a YAML file. (#120299)

Commit
1 year ago
Move torchbench model configuration into a YAML file. (#120299) Summary: This PR moves other aspects of torchbench's model configuration (e.g. batch size, tolerance requirements, etc.) into a new YAML file: `torchbench.yaml`. It also merges the recently added `torchbench_skip_models.yaml` file inside the `skip` key. This is an effort so that external consumers are able to easily replicate the performance results and coverage results from the PyTorch HUD. X-link: https://github.com/pytorch/pytorch/pull/120299 Approved by: https://github.com/jansel Reviewed By: jeanschmidt Differential Revision: D54123721 fbshipit-source-id: c6e69269775fa8a70021fe13313293a527c6b3e1
Author
Parents
Loading