MCPcopy Create free account
hub / github.com/pytorch/pytorch / _zero_grad

Method _zero_grad

test/test_modules.py:322–326  ·  view source on GitHub ↗
(self, obj)

Source from the content-addressed store, hash-verified

320 return self._traverse_obj(obj, inner_get_grad)
321
322 def _zero_grad(self, obj):
323 def inner_zero_grad(obj):
324 if obj.grad is not None:
325 obj.grad = None
326 self._traverse_obj(obj, inner_zero_grad)
327
328 @modules(module_db)
329 def test_non_contiguous_tensors(self, device, dtype, module_info, training):

Callers 1

Calls 1

_traverse_objMethod · 0.95

Tested by

no test coverage detected