validate_model() — pytorch Function Reference

Architecture documentation for the validate_model() function in common.py from the pytorch codebase.

Function python DynamoBenchmarks Configuration calls 3 called by 1

Entity Profile

DynamoBenchmarks→ Configuration→ validate_model() — pytorch Function Reference

Dependency Diagram

graph TD
  684ace56_6d9f_5ea1_f47a_cb8aa5834a49["validate_model()"]
  f731aa1a_ba23_ab06_f8e5_e87d6a842749["load_model()"]
  f731aa1a_ba23_ab06_f8e5_e87d6a842749 -->|calls| 684ace56_6d9f_5ea1_f47a_cb8aa5834a49
  4b936914_ae3d_7efa_4e57_180bdf6f9020["deepcopy_model()"]
  684ace56_6d9f_5ea1_f47a_cb8aa5834a49 -->|calls| 4b936914_ae3d_7efa_4e57_180bdf6f9020
  da8c81f0_c44b_4b89_914d_318c84a98a42["cast_based_on_args()"]
  684ace56_6d9f_5ea1_f47a_cb8aa5834a49 -->|calls| da8c81f0_c44b_4b89_914d_318c84a98a42
  ce8fd365_4112_b289_9c73_7345d5e35203["RuntimeError()"]
  684ace56_6d9f_5ea1_f47a_cb8aa5834a49 -->|calls| ce8fd365_4112_b289_9c73_7345d5e35203
  style 684ace56_6d9f_5ea1_f47a_cb8aa5834a49 fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

benchmarks/dynamo/common.py lines 2027–2037

    def validate_model(self, model, example_inputs):
        """
        Runs the eager model with example inputs to ensure that eager passes.
        """
        model = self.deepcopy_model(model)
        example_inputs = clone_inputs(example_inputs)
        model, example_inputs = self.cast_based_on_args(model, example_inputs)
        try:
            self.model_iter_fn(model, example_inputs)
        except Exception as e:
            raise RuntimeError("Eager run failed") from e