pytorch
7ebc45ea - [dynamo] Better error message for bad timm model name (#91049)

Commit
2 years ago
[dynamo] Better error message for bad timm model name (#91049) Fixes https://github.com/pytorch/torchdynamo/issues/1995 Running `python benchmarks/dynamo/timm_models.py --performance --float32 -dcuda --output=out.csv --training --inductor --only bad_model_name` gives ``` Traceback (most recent call last): File "benchmarks/dynamo/timm_models.py", line 338, in <module> main(TimmRunnner()) File "/scratch/williamwen/work/pytorch/benchmarks/dynamo/common.py", line 1660, in main return maybe_fresh_cache(run, args.cold_start_latency and args.only)( File "/scratch/williamwen/work/pytorch/benchmarks/dynamo/common.py", line 833, in inner return fn(*args, **kwargs) File "/scratch/williamwen/work/pytorch/benchmarks/dynamo/common.py", line 2000, in run ) = runner.load_model(device, model_name, batch_size=batch_size) File "benchmarks/dynamo/timm_models.py", line 215, in load_model raise RuntimeError(f"Failed to load model '{model_name}'") RuntimeError: Failed to load model 'bad_model_name' ``` Pull Request resolved: https://github.com/pytorch/pytorch/pull/91049 Approved by: https://github.com/ezyang
Author
Committer
Parents
Loading