Home / Function/ optimizer_zero_grad() — pytorch Function Reference

optimizer_zero_grad() — pytorch Function Reference

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

Function python DynamoBenchmarks Microbenchmarking called by 1

Entity Profile

Dependency Diagram

graph TD
  237fa410_eacb_e511_495a_212eabaed03a["optimizer_zero_grad()"]
  b66902f6_4c0e_bd3b_c664_55e9f6f9007d["forward_and_backward_pass()"]
  b66902f6_4c0e_bd3b_c664_55e9f6f9007d -->|calls| 237fa410_eacb_e511_495a_212eabaed03a
  style 237fa410_eacb_e511_495a_212eabaed03a fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

benchmarks/dynamo/common.py lines 2077–2081

    def optimizer_zero_grad(self, mod):
        if self.optimizer is not None:
            self.optimizer.zero_grad(True)
        else:
            mod.zero_grad(True)

Domain

Subdomains

Called By

Source

Frequently Asked Questions

What does optimizer_zero_grad() do?
optimizer_zero_grad() is a function in the pytorch codebase.
What calls optimizer_zero_grad()?
optimizer_zero_grad() is called by 1 function(s): forward_and_backward_pass.

Analyze Your Own Codebase

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

Try Supermodel Free