Home / Function/ create_model() — pytorch Function Reference

create_model() — pytorch Function Reference

Architecture documentation for the create_model() function in benchmark.py from the pytorch codebase.

Function python CoreTensor Dispatcher calls 1 called by 2

Entity Profile

Dependency Diagram

graph TD
  72e5e449_391d_8039_3669_200d59371962["create_model()"]
  b3318dd7_fd29_9888_be68_bbac5810e162["create_model()"]
  b3318dd7_fd29_9888_be68_bbac5810e162 -->|calls| 72e5e449_391d_8039_3669_200d59371962
  320a5772_c0de_3469_ef0c_57110d2440fd["benchmark_process_group()"]
  320a5772_c0de_3469_ef0c_57110d2440fd -->|calls| 72e5e449_391d_8039_3669_200d59371962
  b3318dd7_fd29_9888_be68_bbac5810e162["create_model()"]
  72e5e449_391d_8039_3669_200d59371962 -->|calls| b3318dd7_fd29_9888_be68_bbac5810e162
  style 72e5e449_391d_8039_3669_200d59371962 fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

benchmarks/distributed/ddp/benchmark.py lines 187–188

    def create_model(self):
        return torchvision.models.__dict__[self.model]().to(self.device)

Domain

Subdomains

Calls

Called By

Source

Frequently Asked Questions

What does create_model() do?
create_model() is a function in the pytorch codebase.
What does create_model() call?
create_model() calls 1 function(s): create_model.
What calls create_model()?
create_model() is called by 2 function(s): benchmark_process_group, create_model.

Analyze Your Own Codebase

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

Try Supermodel Free