Method count_zeros

codegeex/megatron/optimizer/optimizer.py:85–87 · view source on GitHub ↗

(self)

Source from the content-addressed store, hash-verified

83	return clip_grad_norm_fp32(params, clip_grad)
84
85	def count_zeros(self):
86	params = self.get_parameters()
87	return count_zeros_fp32(params)
88
89	@abstractmethod
90	def zero_grad(self, set_to_none=True):

stepMethod · 0.80

get_parametersMethod · 0.95

count_zeros_fp32Function · 0.85

no test coverage detected