Home / Function/ model_iter_fn() — pytorch Function Reference

model_iter_fn() — pytorch Function Reference

Architecture documentation for the model_iter_fn() function in dist_util.py from the pytorch codebase.

Function python DynamoBenchmarks Microbenchmarking

Entity Profile

Relationship Graph

Source Code

benchmarks/dynamo/dist_util.py lines 80–85

def model_iter_fn(model, example_inputs, collect_outputs=False):
    outputs = model(*example_inputs)
    loss = reduce_to_scalar_loss(outputs)
    loss.backward()
    if collect_outputs:
        return outputs

Domain

Subdomains

Source

Analyze Your Own Codebase

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

Try Supermodel Free