MCPcopy Create free account
hub / github.com/TeleHuman/TextOp / zero_master_grads

Function zero_master_grads

TextOpRobotMDAR/robotmdar/diffusion/fp16_util.py:128–130  ·  view source on GitHub ↗
(master_params)

Source from the content-addressed store, hash-verified

126
127
128def zero_master_grads(master_params):
129 for param in master_params:
130 param.grad = None
131
132
133def zero_grad(model_params):

Callers 1

_optimize_fp16Method · 0.85

Calls

no outgoing calls

Tested by

no test coverage detected