Home / Function/ args() — pytorch Function Reference

args() — pytorch Function Reference

Architecture documentation for the args() function in benchmarks.py from the pytorch codebase.

Entity Profile

Dependency Diagram

graph TD
  815a46f9_d548_98bf_cbd7_cb9fae326824["args()"]
  f6997f23_fa92_edba_2dc0_4e174f309eb9["parse_args()"]
  815a46f9_d548_98bf_cbd7_cb9fae326824 -->|calls| f6997f23_fa92_edba_2dc0_4e174f309eb9
  b93ae0ef_d85a_81fb_b07f_38119c7d07cc["huggingface_main()"]
  815a46f9_d548_98bf_cbd7_cb9fae326824 -->|calls| b93ae0ef_d85a_81fb_b07f_38119c7d07cc
  610c17b4_1930_ccc8_d57d_185cea285f5b["print()"]
  815a46f9_d548_98bf_cbd7_cb9fae326824 -->|calls| 610c17b4_1930_ccc8_d57d_185cea285f5b
  style 815a46f9_d548_98bf_cbd7_cb9fae326824 fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

benchmarks/dynamo/benchmarks.py lines 90–119

    args, unknown = parse_args()
    if args.only:
        name = args.only
        if name in TIMM_MODEL_NAMES:
            import timm_models

            timm_models.timm_main()
        elif name in HF_MODELS_FILE_NAME:
            import huggingface

            huggingface.huggingface_main()
        elif name in TORCHBENCH_MODELS_FILE_NAME:
            import torchbench

            torchbench.torchbench_main()
        else:
            print(f"Illegal model name? {name}")
            sys.exit(-1)
    else:
        import torchbench

        torchbench.torchbench_main()

        import huggingface

        huggingface.huggingface_main()

        import timm_models

        timm_models.timm_main()

Subdomains

Frequently Asked Questions

What does args() do?
args() is a function in the pytorch codebase.
What does args() call?
args() calls 3 function(s): huggingface_main, parse_args, print.

Analyze Your Own Codebase

Get architecture documentation, dependency graphs, and domain analysis for your codebase in minutes.

Try Supermodel Free